Home | History | Annotate | Download | only in utils
      1 /*
      2  * Radiotap parser
      3  *
      4  * Copyright 2007		Andy Green <andy (at) warmcat.com>
      5  * Copyright 2009		Johannes Berg <johannes (at) 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 "platform.h"
     17 #include "radiotap_iter.h"
     18 
     19 /* function prototypes and related defs are in radiotap_iter.h */
     20 
     21 static const struct radiotap_align_size rtap_namespace_sizes[] = {
     22 	[IEEE80211_RADIOTAP_TSFT] = { .align = 8, .size = 8, },
     23 	[IEEE80211_RADIOTAP_FLAGS] = { .align = 1, .size = 1, },
     24 	[IEEE80211_RADIOTAP_RATE] = { .align = 1, .size = 1, },
     25 	[IEEE80211_RADIOTAP_CHANNEL] = { .align = 2, .size = 4, },
     26 	[IEEE80211_RADIOTAP_FHSS] = { .align = 2, .size = 2, },
     27 	[IEEE80211_RADIOTAP_DBM_ANTSIGNAL] = { .align = 1, .size = 1, },
     28 	[IEEE80211_RADIOTAP_DBM_ANTNOISE] = { .align = 1, .size = 1, },
     29 	[IEEE80211_RADIOTAP_LOCK_QUALITY] = { .align = 2, .size = 2, },
     30 	[IEEE80211_RADIOTAP_TX_ATTENUATION] = { .align = 2, .size = 2, },
     31 	[IEEE80211_RADIOTAP_DB_TX_ATTENUATION] = { .align = 2, .size = 2, },
     32 	[IEEE80211_RADIOTAP_DBM_TX_POWER] = { .align = 1, .size = 1, },
     33 	[IEEE80211_RADIOTAP_ANTENNA] = { .align = 1, .size = 1, },
     34 	[IEEE80211_RADIOTAP_DB_ANTSIGNAL] = { .align = 1, .size = 1, },
     35 	[IEEE80211_RADIOTAP_DB_ANTNOISE] = { .align = 1, .size = 1, },
     36 	[IEEE80211_RADIOTAP_RX_FLAGS] = { .align = 2, .size = 2, },
     37 	[IEEE80211_RADIOTAP_TX_FLAGS] = { .align = 2, .size = 2, },
     38 	[IEEE80211_RADIOTAP_RTS_RETRIES] = { .align = 1, .size = 1, },
     39 	[IEEE80211_RADIOTAP_DATA_RETRIES] = { .align = 1, .size = 1, },
     40 	[IEEE80211_RADIOTAP_MCS] = { .align = 1, .size = 3, },
     41 	[IEEE80211_RADIOTAP_AMPDU_STATUS] = { .align = 4, .size = 8, },
     42 	/*
     43 	 * add more here as they are defined in radiotap.h
     44 	 */
     45 };
     46 
     47 static const struct ieee80211_radiotap_namespace radiotap_ns = {
     48 	.n_bits = sizeof(rtap_namespace_sizes) / sizeof(rtap_namespace_sizes[0]),
     49 	.align_size = rtap_namespace_sizes,
     50 };
     51 
     52 /**
     53  * ieee80211_radiotap_iterator_init - radiotap parser iterator initialization
     54  * @iterator: radiotap_iterator to initialize
     55  * @radiotap_header: radiotap header to parse
     56  * @max_length: total length we can parse into (eg, whole packet length)
     57  *
     58  * Returns: 0 or a negative error code if there is a problem.
     59  *
     60  * This function initializes an opaque iterator struct which can then
     61  * be passed to ieee80211_radiotap_iterator_next() to visit every radiotap
     62  * argument which is present in the header.  It knows about extended
     63  * present headers and handles them.
     64  *
     65  * How to use:
     66  * call __ieee80211_radiotap_iterator_init() to init a semi-opaque iterator
     67  * struct ieee80211_radiotap_iterator (no need to init the struct beforehand)
     68  * checking for a good 0 return code.  Then loop calling
     69  * __ieee80211_radiotap_iterator_next()... it returns either 0,
     70  * -ENOENT if there are no more args to parse, or -EINVAL if there is a problem.
     71  * The iterator's @this_arg member points to the start of the argument
     72  * associated with the current argument index that is present, which can be
     73  * found in the iterator's @this_arg_index member.  This arg index corresponds
     74  * to the IEEE80211_RADIOTAP_... defines.
     75  *
     76  * Radiotap header length:
     77  * You can find the CPU-endian total radiotap header length in
     78  * iterator->max_length after executing ieee80211_radiotap_iterator_init()
     79  * successfully.
     80  *
     81  * Alignment Gotcha:
     82  * You must take care when dereferencing iterator.this_arg
     83  * for multibyte types... the pointer is not aligned.  Use
     84  * get_unaligned((type *)iterator.this_arg) to dereference
     85  * iterator.this_arg for type "type" safely on all arches.
     86  *
     87  * Example code: parse.c
     88  */
     89 
     90 int ieee80211_radiotap_iterator_init(
     91 	struct ieee80211_radiotap_iterator *iterator,
     92 	struct ieee80211_radiotap_header *radiotap_header,
     93 	int max_length, const struct ieee80211_radiotap_vendor_namespaces *vns)
     94 {
     95 	/* must at least have the radiotap header */
     96 	if (max_length < (int)sizeof(struct ieee80211_radiotap_header))
     97 		return -EINVAL;
     98 
     99 	/* Linux only supports version 0 radiotap format */
    100 	if (radiotap_header->it_version)
    101 		return -EINVAL;
    102 
    103 	/* sanity check for allowed length and radiotap length field */
    104 	if (max_length < get_unaligned_le16(&radiotap_header->it_len))
    105 		return -EINVAL;
    106 
    107 	iterator->_rtheader = radiotap_header;
    108 	iterator->_max_length = get_unaligned_le16(&radiotap_header->it_len);
    109 	iterator->_arg_index = 0;
    110 	iterator->_bitmap_shifter = get_unaligned_le32(&radiotap_header->it_present);
    111 	iterator->_arg = (uint8_t *)radiotap_header + sizeof(*radiotap_header);
    112 	iterator->_next_ns_data = NULL;
    113 	iterator->_reset_on_ext = 0;
    114 	iterator->_next_bitmap = &radiotap_header->it_present;
    115 	iterator->_next_bitmap++;
    116 	iterator->_vns = vns;
    117 	iterator->current_namespace = &radiotap_ns;
    118 	iterator->is_radiotap_ns = 1;
    119 #ifdef RADIOTAP_SUPPORT_OVERRIDES
    120 	iterator->n_overrides = 0;
    121 	iterator->overrides = NULL;
    122 #endif
    123 
    124 	/* find payload start allowing for extended bitmap(s) */
    125 
    126 	if (iterator->_bitmap_shifter & BIT(IEEE80211_RADIOTAP_EXT)) {
    127 		if ((unsigned long)iterator->_arg -
    128 		    (unsigned long)iterator->_rtheader + sizeof(uint32_t) >
    129 		    (unsigned long)iterator->_max_length)
    130 			return -EINVAL;
    131 		while (get_unaligned_le32(iterator->_arg) &
    132 		       BIT(IEEE80211_RADIOTAP_EXT)) {
    133 			iterator->_arg += sizeof(uint32_t);
    134 
    135 			/*
    136 			 * check for insanity where the present bitmaps
    137 			 * keep claiming to extend up to or even beyond the
    138 			 * stated radiotap header length
    139 			 */
    140 
    141 			if ((unsigned long)iterator->_arg -
    142 			    (unsigned long)iterator->_rtheader +
    143 			    sizeof(uint32_t) >
    144 			    (unsigned long)iterator->_max_length)
    145 				return -EINVAL;
    146 		}
    147 
    148 		iterator->_arg += sizeof(uint32_t);
    149 
    150 		/*
    151 		 * no need to check again for blowing past stated radiotap
    152 		 * header length, because ieee80211_radiotap_iterator_next
    153 		 * checks it before it is dereferenced
    154 		 */
    155 	}
    156 
    157 	iterator->this_arg = iterator->_arg;
    158 	iterator->this_arg_index = 0;
    159 	iterator->this_arg_size = 0;
    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 }
    397