Added wifi sniffer basics
[soho-sigint.git] / client-wifi / disect5 / radiotap.c
1 /*
2  * Radiotap parser
3  *
4  * Copyright 2007               Andy Green <andy@warmcat.com>
5  * Copyright 2009               Johannes Berg <johannes@sipsolutions.net>
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  * Alternatively, this software may be distributed under the terms of BSD
12  * license.
13  *
14  * See COPYING for more details.
15  */
16 #include "radiotap_iter.h"
17 #include "platform.h"
18 #if defined(ANDROID) || defined(__ANDROID__)
19         #include "../byteorder.h"
20 #endif
21
22 /* function prototypes and related defs are in radiotap_iter.h */
23
24 static const struct radiotap_align_size rtap_namespace_sizes[] = {
25         [IEEE80211_RADIOTAP_TSFT] = { .align = 8, .size = 8, },
26         [IEEE80211_RADIOTAP_FLAGS] = { .align = 1, .size = 1, },
27         [IEEE80211_RADIOTAP_RATE] = { .align = 1, .size = 1, },
28         [IEEE80211_RADIOTAP_CHANNEL] = { .align = 2, .size = 4, },
29         [IEEE80211_RADIOTAP_FHSS] = { .align = 2, .size = 2, },
30         [IEEE80211_RADIOTAP_DBM_ANTSIGNAL] = { .align = 1, .size = 1, },
31         [IEEE80211_RADIOTAP_DBM_ANTNOISE] = { .align = 1, .size = 1, },
32         [IEEE80211_RADIOTAP_LOCK_QUALITY] = { .align = 2, .size = 2, },
33         [IEEE80211_RADIOTAP_TX_ATTENUATION] = { .align = 2, .size = 2, },
34         [IEEE80211_RADIOTAP_DB_TX_ATTENUATION] = { .align = 2, .size = 2, },
35         [IEEE80211_RADIOTAP_DBM_TX_POWER] = { .align = 1, .size = 1, },
36         [IEEE80211_RADIOTAP_ANTENNA] = { .align = 1, .size = 1, },
37         [IEEE80211_RADIOTAP_DB_ANTSIGNAL] = { .align = 1, .size = 1, },
38         [IEEE80211_RADIOTAP_DB_ANTNOISE] = { .align = 1, .size = 1, },
39         [IEEE80211_RADIOTAP_RX_FLAGS] = { .align = 2, .size = 2, },
40         [IEEE80211_RADIOTAP_TX_FLAGS] = { .align = 2, .size = 2, },
41         [IEEE80211_RADIOTAP_RTS_RETRIES] = { .align = 1, .size = 1, },
42         [IEEE80211_RADIOTAP_DATA_RETRIES] = { .align = 1, .size = 1, },
43         [IEEE80211_RADIOTAP_MCS] = { .align = 1, .size = 3, },
44         [IEEE80211_RADIOTAP_AMPDU_STATUS] = { .align = 4, .size = 8, },
45         /*
46          * add more here as they are defined in radiotap.h
47          */
48 };
49
50 static const struct ieee80211_radiotap_namespace radiotap_ns = {
51         .n_bits = sizeof(rtap_namespace_sizes) / sizeof(rtap_namespace_sizes[0]),
52         .align_size = rtap_namespace_sizes,
53 };
54
55 /**
56  * ieee80211_radiotap_iterator_init - radiotap parser iterator initialization
57  * @iterator: radiotap_iterator to initialize
58  * @radiotap_header: radiotap header to parse
59  * @max_length: total length we can parse into (eg, whole packet length)
60  *
61  * Returns: 0 or a negative error code if there is a problem.
62  *
63  * This function initializes an opaque iterator struct which can then
64  * be passed to ieee80211_radiotap_iterator_next() to visit every radiotap
65  * argument which is present in the header.  It knows about extended
66  * present headers and handles them.
67  *
68  * How to use:
69  * call __ieee80211_radiotap_iterator_init() to init a semi-opaque iterator
70  * struct ieee80211_radiotap_iterator (no need to init the struct beforehand)
71  * checking for a good 0 return code.  Then loop calling
72  * __ieee80211_radiotap_iterator_next()... it returns either 0,
73  * -ENOENT if there are no more args to parse, or -EINVAL if there is a problem.
74  * The iterator's @this_arg member points to the start of the argument
75  * associated with the current argument index that is present, which can be
76  * found in the iterator's @this_arg_index member.  This arg index corresponds
77  * to the IEEE80211_RADIOTAP_... defines.
78  *
79  * Radiotap header length:
80  * You can find the CPU-endian total radiotap header length in
81  * iterator->max_length after executing ieee80211_radiotap_iterator_init()
82  * successfully.
83  *
84  * Alignment Gotcha:
85  * You must take care when dereferencing iterator.this_arg
86  * for multibyte types... the pointer is not aligned.  Use
87  * get_unaligned((type *)iterator.this_arg) to dereference
88  * iterator.this_arg for type "type" safely on all arches.
89  *
90  * Example code: parse.c
91  */
92
93 int ieee80211_radiotap_iterator_init(
94         struct ieee80211_radiotap_iterator *iterator,
95         struct ieee80211_radiotap_header *radiotap_header,
96         int max_length, const struct ieee80211_radiotap_vendor_namespaces *vns)
97 {
98         /* must at least have the radiotap header */
99         if (max_length < (int)sizeof(struct ieee80211_radiotap_header))
100                 return -EINVAL;
101
102         /* Linux only supports version 0 radiotap format */
103         if (radiotap_header->it_version)
104                 return -EINVAL;
105
106         /* sanity check for allowed length and radiotap length field */
107         if (max_length < get_unaligned_le16(&radiotap_header->it_len))
108                 return -EINVAL;
109
110         iterator->_rtheader = radiotap_header;
111         iterator->_max_length = get_unaligned_le16(&radiotap_header->it_len);
112         iterator->_arg_index = 0;
113         iterator->_bitmap_shifter = get_unaligned_le32(&radiotap_header->it_present);
114         iterator->_arg = (uint8_t *)radiotap_header + sizeof(*radiotap_header);
115         iterator->_reset_on_ext = 0;
116         iterator->_next_bitmap = &radiotap_header->it_present;
117         iterator->_next_bitmap++;
118         iterator->_vns = vns;
119         iterator->current_namespace = &radiotap_ns;
120         iterator->is_radiotap_ns = 1;
121 #ifdef RADIOTAP_SUPPORT_OVERRIDES
122         iterator->n_overrides = 0;
123         iterator->overrides = NULL;
124 #endif
125
126         /* find payload start allowing for extended bitmap(s) */
127
128         if (iterator->_bitmap_shifter & (1<<IEEE80211_RADIOTAP_EXT)) {
129                 if ((unsigned long)iterator->_arg -
130                     (unsigned long)iterator->_rtheader + sizeof(uint32_t) >
131                     (unsigned long)iterator->_max_length)
132                         return -EINVAL;
133                 while (get_unaligned_le32(iterator->_arg) &
134                                         (1 << IEEE80211_RADIOTAP_EXT)) {
135                         iterator->_arg += sizeof(uint32_t);
136
137                         /*
138                          * check for insanity where the present bitmaps
139                          * keep claiming to extend up to or even beyond the
140                          * stated radiotap header length
141                          */
142
143                         if ((unsigned long)iterator->_arg -
144                             (unsigned long)iterator->_rtheader +
145                             sizeof(uint32_t) >
146                             (unsigned long)iterator->_max_length)
147                                 return -EINVAL;
148                 }
149
150                 iterator->_arg += sizeof(uint32_t);
151
152                 /*
153                  * no need to check again for blowing past stated radiotap
154                  * header length, because ieee80211_radiotap_iterator_next
155                  * checks it before it is dereferenced
156                  */
157         }
158
159         iterator->this_arg = iterator->_arg;
160
161         /* we are all initialized happily */
162
163         return 0;
164 }
165
166 static void find_ns(struct ieee80211_radiotap_iterator *iterator,
167                     uint32_t oui, uint8_t subns)
168 {
169         int i;
170
171         iterator->current_namespace = NULL;
172
173         if (!iterator->_vns)
174                 return;
175
176         for (i = 0; i < iterator->_vns->n_ns; i++) {
177                 if (iterator->_vns->ns[i].oui != oui)
178                         continue;
179                 if (iterator->_vns->ns[i].subns != subns)
180                         continue;
181
182                 iterator->current_namespace = &iterator->_vns->ns[i];
183                 break;
184         }
185 }
186
187 #ifdef RADIOTAP_SUPPORT_OVERRIDES
188 static int find_override(struct ieee80211_radiotap_iterator *iterator,
189                          int *align, int *size)
190 {
191         int i;
192
193         if (!iterator->overrides)
194                 return 0;
195
196         for (i = 0; i < iterator->n_overrides; i++) {
197                 if (iterator->_arg_index == iterator->overrides[i].field) {
198                         *align = iterator->overrides[i].align;
199                         *size = iterator->overrides[i].size;
200                         if (!*align) /* erroneous override */
201                                 return 0;
202                         return 1;
203                 }
204         }
205
206         return 0;
207 }
208 #endif
209
210
211 /**
212  * ieee80211_radiotap_iterator_next - return next radiotap parser iterator arg
213  * @iterator: radiotap_iterator to move to next arg (if any)
214  *
215  * Returns: 0 if there is an argument to handle,
216  * -ENOENT if there are no more args or -EINVAL
217  * if there is something else wrong.
218  *
219  * This function provides the next radiotap arg index (IEEE80211_RADIOTAP_*)
220  * in @this_arg_index and sets @this_arg to point to the
221  * payload for the field.  It takes care of alignment handling and extended
222  * present fields.  @this_arg can be changed by the caller (eg,
223  * incremented to move inside a compound argument like
224  * IEEE80211_RADIOTAP_CHANNEL).  The args pointed to are in
225  * little-endian format whatever the endianess of your CPU.
226  *
227  * Alignment Gotcha:
228  * You must take care when dereferencing iterator.this_arg
229  * for multibyte types... the pointer is not aligned.  Use
230  * get_unaligned((type *)iterator.this_arg) to dereference
231  * iterator.this_arg for type "type" safely on all arches.
232  */
233
234 int ieee80211_radiotap_iterator_next(
235         struct ieee80211_radiotap_iterator *iterator)
236 {
237         while (1) {
238                 int hit = 0;
239                 int pad, align, size, subns;
240                 uint32_t oui;
241
242                 /* if no more EXT bits, that's it */
243                 if ((iterator->_arg_index % 32) == IEEE80211_RADIOTAP_EXT &&
244                     !(iterator->_bitmap_shifter & 1))
245                         return -ENOENT;
246
247                 if (!(iterator->_bitmap_shifter & 1))
248                         goto next_entry; /* arg not present */
249
250                 /* get alignment/size of data */
251                 switch (iterator->_arg_index % 32) {
252                 case IEEE80211_RADIOTAP_RADIOTAP_NAMESPACE:
253                 case IEEE80211_RADIOTAP_EXT:
254                         align = 1;
255                         size = 0;
256                         break;
257                 case IEEE80211_RADIOTAP_VENDOR_NAMESPACE:
258                         align = 2;
259                         size = 6;
260                         break;
261                 default:
262 #ifdef RADIOTAP_SUPPORT_OVERRIDES
263                         if (find_override(iterator, &align, &size)) {
264                                 /* all set */
265                         } else
266 #endif
267                         if (!iterator->current_namespace ||
268                             iterator->_arg_index >= iterator->current_namespace->n_bits) {
269                                 if (iterator->current_namespace == &radiotap_ns)
270                                         return -ENOENT;
271                                 align = 0;
272                         } else {
273                                 align = iterator->current_namespace->align_size[iterator->_arg_index].align;
274                                 size = iterator->current_namespace->align_size[iterator->_arg_index].size;
275                         }
276                         if (!align) {
277                                 /* skip all subsequent data */
278                                 iterator->_arg = iterator->_next_ns_data;
279                                 /* give up on this namespace */
280                                 iterator->current_namespace = NULL;
281                                 goto next_entry;
282                         }
283                         break;
284                 }
285
286                 /*
287                  * arg is present, account for alignment padding
288                  *
289                  * Note that these alignments are relative to the start
290                  * of the radiotap header.  There is no guarantee
291                  * that the radiotap header itself is aligned on any
292                  * kind of boundary.
293                  *
294                  * The above is why get_unaligned() is used to dereference
295                  * multibyte elements from the radiotap area.
296                  */
297
298                 pad = ((unsigned long)iterator->_arg -
299                        (unsigned long)iterator->_rtheader) & (align - 1);
300
301                 if (pad)
302                         iterator->_arg += align - pad;
303
304                 if (iterator->_arg_index % 32 == IEEE80211_RADIOTAP_VENDOR_NAMESPACE) {
305                         int vnslen;
306
307                         if ((unsigned long)iterator->_arg + size -
308                             (unsigned long)iterator->_rtheader >
309                             (unsigned long)iterator->_max_length)
310                                 return -EINVAL;
311
312                         oui = (*iterator->_arg << 16) |
313                                 (*(iterator->_arg + 1) << 8) |
314                                 *(iterator->_arg + 2);
315                         subns = *(iterator->_arg + 3);
316
317                         find_ns(iterator, oui, subns);
318
319                         vnslen = get_unaligned_le16(iterator->_arg + 4);
320                         iterator->_next_ns_data = iterator->_arg + size + vnslen;
321                         if (!iterator->current_namespace)
322                                 size += vnslen;
323                 }
324
325                 /*
326                  * this is what we will return to user, but we need to
327                  * move on first so next call has something fresh to test
328                  */
329                 iterator->this_arg_index = iterator->_arg_index;
330                 iterator->this_arg = iterator->_arg;
331                 iterator->this_arg_size = size;
332
333                 /* internally move on the size of this arg */
334                 iterator->_arg += size;
335
336                 /*
337                  * check for insanity where we are given a bitmap that
338                  * claims to have more arg content than the length of the
339                  * radiotap section.  We will normally end up equalling this
340                  * max_length on the last arg, never exceeding it.
341                  */
342
343                 if ((unsigned long)iterator->_arg -
344                     (unsigned long)iterator->_rtheader >
345                     (unsigned long)iterator->_max_length)
346                         return -EINVAL;
347
348                 /* these special ones are valid in each bitmap word */
349                 switch (iterator->_arg_index % 32) {
350                 case IEEE80211_RADIOTAP_VENDOR_NAMESPACE:
351                         iterator->_reset_on_ext = 1;
352
353                         iterator->is_radiotap_ns = 0;
354                         /*
355                          * If parser didn't register this vendor
356                          * namespace with us, allow it to show it
357                          * as 'raw. Do do that, set argument index
358                          * to vendor namespace.
359                          */
360                         iterator->this_arg_index =
361                                 IEEE80211_RADIOTAP_VENDOR_NAMESPACE;
362                         if (!iterator->current_namespace)
363                                 hit = 1;
364                         goto next_entry;
365                 case IEEE80211_RADIOTAP_RADIOTAP_NAMESPACE:
366                         iterator->_reset_on_ext = 1;
367                         iterator->current_namespace = &radiotap_ns;
368                         iterator->is_radiotap_ns = 1;
369                         goto next_entry;
370                 case IEEE80211_RADIOTAP_EXT:
371                         /*
372                          * bit 31 was set, there is more
373                          * -- move to next u32 bitmap
374                          */
375                         iterator->_bitmap_shifter =
376                                 get_unaligned_le32(iterator->_next_bitmap);
377                         iterator->_next_bitmap++;
378                         if (iterator->_reset_on_ext)
379                                 iterator->_arg_index = 0;
380                         else
381                                 iterator->_arg_index++;
382                         iterator->_reset_on_ext = 0;
383                         break;
384                 default:
385                         /* we've got a hit! */
386                         hit = 1;
387  next_entry:
388                         iterator->_bitmap_shifter >>= 1;
389                         iterator->_arg_index++;
390                 }
391
392                 /* if we found a valid arg earlier, return it now */
393                 if (hit)
394                         return 0;
395         }
396 }