diff options
Diffstat (limited to 'scripts/external_libs/dpkt-1.8.6.2/dpkt/radiotap.py')
| -rw-r--r-- | scripts/external_libs/dpkt-1.8.6.2/dpkt/radiotap.py | 532 |
1 files changed, 0 insertions, 532 deletions
diff --git a/scripts/external_libs/dpkt-1.8.6.2/dpkt/radiotap.py b/scripts/external_libs/dpkt-1.8.6.2/dpkt/radiotap.py deleted file mode 100644 index 76e6306b..00000000 --- a/scripts/external_libs/dpkt-1.8.6.2/dpkt/radiotap.py +++ /dev/null @@ -1,532 +0,0 @@ -# -*- coding: utf-8 -*- -"""Radiotap""" - -import dpkt -import ieee80211 -import socket -from decorators import deprecated - -# Ref: http://www.radiotap.org -# Fields Ref: http://www.radiotap.org/defined-fields/all - -# Present flags -_TSFT_MASK = 0x1000000 -_FLAGS_MASK = 0x2000000 -_RATE_MASK = 0x4000000 -_CHANNEL_MASK = 0x8000000 -_FHSS_MASK = 0x10000000 -_ANT_SIG_MASK = 0x20000000 -_ANT_NOISE_MASK = 0x40000000 -_LOCK_QUAL_MASK = 0x80000000 -_TX_ATTN_MASK = 0x10000 -_DB_TX_ATTN_MASK = 0x20000 -_DBM_TX_POWER_MASK = 0x40000 -_ANTENNA_MASK = 0x80000 -_DB_ANT_SIG_MASK = 0x100000 -_DB_ANT_NOISE_MASK = 0x200000 -_RX_FLAGS_MASK = 0x400000 -_CHANNELPLUS_MASK = 0x200 -_EXT_MASK = 0x1 - -_TSFT_SHIFT = 24 -_FLAGS_SHIFT = 25 -_RATE_SHIFT = 26 -_CHANNEL_SHIFT = 27 -_FHSS_SHIFT = 28 -_ANT_SIG_SHIFT = 29 -_ANT_NOISE_SHIFT = 30 -_LOCK_QUAL_SHIFT = 31 -_TX_ATTN_SHIFT = 16 -_DB_TX_ATTN_SHIFT = 17 -_DBM_TX_POWER_SHIFT = 18 -_ANTENNA_SHIFT = 19 -_DB_ANT_SIG_SHIFT = 20 -_DB_ANT_NOISE_SHIFT = 21 -_RX_FLAGS_SHIFT = 22 -_CHANNELPLUS_SHIFT = 10 -_EXT_SHIFT = 0 - -# Flags elements -_FLAGS_SIZE = 2 -_CFP_FLAG_SHIFT = 0 -_PREAMBLE_SHIFT = 1 -_WEP_SHIFT = 2 -_FRAG_SHIFT = 3 -_FCS_SHIFT = 4 -_DATA_PAD_SHIFT = 5 -_BAD_FCS_SHIFT = 6 -_SHORT_GI_SHIFT = 7 - -# Channel type -_CHAN_TYPE_SIZE = 4 -_CHANNEL_TYPE_SHIFT = 4 -_CCK_SHIFT = 5 -_OFDM_SHIFT = 6 -_TWO_GHZ_SHIFT = 7 -_FIVE_GHZ_SHIFT = 8 -_PASSIVE_SHIFT = 9 -_DYN_CCK_OFDM_SHIFT = 10 -_GFSK_SHIFT = 11 -_GSM_SHIFT = 12 -_STATIC_TURBO_SHIFT = 13 -_HALF_RATE_SHIFT = 14 -_QUARTER_RATE_SHIFT = 15 - -# Flags offsets and masks -_FCS_SHIFT = 4 -_FCS_MASK = 0x10 - - -class Radiotap(dpkt.Packet): - __hdr__ = ( - ('version', 'B', 0), - ('pad', 'B', 0), - ('length', 'H', 0), - ('present_flags', 'I', 0) - ) - - @property - def tsft_present(self): - return (self.present_flags & _TSFT_MASK) >> _TSFT_SHIFT - - @tsft_present.setter - def tsft_present(self, val): - self.present_flags |= val << _TSFT_SHIFT - - @property - def flags_present(self): - return (self.present_flags & _FLAGS_MASK) >> _FLAGS_SHIFT - - @flags_present.setter - def flags_present(self, val): - self.present_flags |= val << _FLAGS_SHIFT - - @property - def rate_present(self): - return (self.present_flags & _RATE_MASK) >> _RATE_SHIFT - - @rate_present.setter - def rate_present(self, val): - self.present_flags |= val << _RATE_SHIFT - - @property - def channel_present(self): - return (self.present_flags & _CHANNEL_MASK) >> _CHANNEL_SHIFT - - @channel_present.setter - def channel_present(self, val): - self.present_flags |= val << _CHANNEL_SHIFT - - @property - def fhss_present(self): - return (self.present_flags & _FHSS_MASK) >> _FHSS_SHIFT - - @fhss_present.setter - def fhss_present(self, val): - self.present_flags |= val << _FHSS_SHIFT - - @property - def ant_sig_present(self): - return (self.present_flags & _ANT_SIG_MASK) >> _ANT_SIG_SHIFT - - @ant_sig_present.setter - def ant_sig_present(self, val): - self.present_flags |= val << _ANT_SIG_SHIFT - - @property - def ant_noise_present(self): - return (self.present_flags & _ANT_NOISE_MASK) >> _ANT_NOISE_SHIFT - - @ant_noise_present.setter - def ant_noise_present(self, val): - self.present_flags |= val << _ANT_NOISE_SHIFT - - @property - def lock_qual_present(self): - return (self.present_flags & _LOCK_QUAL_MASK) >> _LOCK_QUAL_SHIFT - - @lock_qual_present.setter - def lock_qual_present(self, val): - self.present_flags |= val << _LOCK_QUAL_SHIFT - - @property - def tx_attn_present(self): - return (self.present_flags & _TX_ATTN_MASK) >> _TX_ATTN_SHIFT - - @tx_attn_present.setter - def tx_attn_present(self, val): - self.present_flags |= val << _TX_ATTN_SHIFT - - @property - def db_tx_attn_present(self): - return (self.present_flags & _DB_TX_ATTN_MASK) >> _DB_TX_ATTN_SHIFT - - @db_tx_attn_present.setter - def db_tx_attn_present(self, val): - self.present_flags |= val << _DB_TX_ATTN_SHIFT - - @property - def dbm_tx_power_present(self): - return (self.present_flags & _DBM_TX_POWER_MASK) >> _DBM_TX_POWER_SHIFT - - @dbm_tx_power_present.setter - def dbm_tx_power_present(self, val): - self.present_flags |= val << _DBM_TX_POWER_SHIFT - - @property - def ant_present(self): - return (self.present_flags & _ANTENNA_MASK) >> _ANTENNA_SHIFT - - @ant_present.setter - def ant_present(self, val): - self.present_flags |= val << _ANTENNA_SHIFT - - @property - def db_ant_sig_present(self): - return (self.present_flags & _DB_ANT_SIG_MASK) >> _DB_ANT_SIG_SHIFT - - @db_ant_sig_present.setter - def db_ant_sig_present(self, val): - self.present_flags |= val << _DB_ANT_SIG_SHIFT - - @property - def db_ant_noise_present(self): - return (self.present_flags & _DB_ANT_NOISE_MASK) >> _DB_ANT_NOISE_SHIFT - - @db_ant_noise_present.setter - def db_ant_noise_present(self, val): - self.present_flags |= val << _DB_ANT_NOISE_SHIFT - - @property - def rx_flags_present(self): - return (self.present_flags & _RX_FLAGS_MASK) >> _RX_FLAGS_SHIFT - - @rx_flags_present.setter - def rx_flags_present(self, val): - self.present_flags |= val << _RX_FLAGS_SHIFT - - @property - def chanplus_present(self): - return (self.present_flags & _CHANNELPLUS_MASK) >> _CHANNELPLUS_SHIFT - - @chanplus_present.setter - def chanplus_present(self, val): - self.present_flags |= val << _CHANNELPLUS_SHIFT - - @property - def ext_present(self): - return (self.present_flags & _EXT_MASK) >> _EXT_SHIFT - - @ext_present.setter - def ext_present(self, val): - self.present_flags |= val << _EXT_SHIFT - - # Deprecated methods, will be removed in the future - # ================================================= - @deprecated - def _get_tsft_present(self): - return self.tsft_present - - @deprecated - def _set_tsft_present(self, val): - self.tsft_present = val - - @deprecated - def _get_flags_present(self): - return self.flags_present - - @deprecated - def _set_flags_present(self, val): - self.flags_present = val - - @deprecated - def _get_rate_present(self): - return self.rate_present - - @deprecated - def _set_rate_present(self, val): - self.rate_present = val - - @deprecated - def _get_channel_present(self): - return self.channel_present - - @deprecated - def _set_channel_present(self, val): - self.channel_present = val - - @deprecated - def _get_fhss_present(self): - return self.fhss_present - - @deprecated - def _set_fhss_present(self, val): - self.fhss_present = val - - @deprecated - def _get_ant_sig_present(self): - return self.ant_sig_present - - @deprecated - def _set_ant_sig_present(self, val): - self.ant_sig_present = val - - @deprecated - def _get_ant_noise_present(self): - return self.ant_noise_present - - @deprecated - def _set_ant_noise_present(self, val): - self.ant_noise_present = val - - @deprecated - def _get_lock_qual_present(self): - return self.lock_qual_present - - @deprecated - def _set_lock_qual_present(self, val): - self.lock_qual_present = val - - @deprecated - def _get_tx_attn_present(self): - return self.tx_attn_present - - @deprecated - def _set_tx_attn_present(self, val): - self.tx_attn_present = val - - @deprecated - def _get_db_tx_attn_present(self): - return self.db_tx_attn_present - - @deprecated - def _set_db_tx_attn_present(self, val): - self.db_tx_attn_present = val - - @deprecated - def _get_dbm_power_present(self): - return self.dbm_tx_power_present - - @deprecated - def _set_dbm_power_present(self, val): - self.dbm_tx_power_present = val - - @deprecated - def _get_ant_present(self): - return self.ant_present - - @deprecated - def _set_ant_present(self, val): - self.ant_present = val - - @deprecated - def _get_db_ant_sig_present(self): - return self.db_ant_sig_present - - @deprecated - def _set_db_ant_sig_present(self, val): - self.db_ant_sig_present = val - - @deprecated - def _get_db_ant_noise_present(self): - return self.db_ant_noise_present - - @deprecated - def _set_db_ant_noise_present(self, val): - self.db_ant_noise_present = val - - @deprecated - def _get_rx_flags_present(self): - return self.rx_flags_present - - @deprecated - def _set_rx_flags_present(self, val): - self.rx_flags_present = val - - @deprecated - def _get_chanplus_present(self): - return self.chanplus_present - - @deprecated - def _set_chanplus_present(self, val): - self.chanplus_present = val - - @deprecated - def _get_ext_present(self): - return self.ext_present - - @deprecated - def _set_ext_present(self, val): - self.ext_present = val - - # ================================================= - - def unpack(self, buf): - dpkt.Packet.unpack(self, buf) - self.data = buf[socket.ntohs(self.length):] - - self.fields = [] - buf = buf[self.__hdr_len__:] - - # decode each field into self.<name> (eg. self.tsft) as well as append it self.fields list - field_decoder = [ - ('tsft', self.tsft_present, self.TSFT), - ('flags', self.flags_present, self.Flags), - ('rate', self.rate_present, self.Rate), - ('channel', self.channel_present, self.Channel), - ('fhss', self.fhss_present, self.FHSS), - ('ant_sig', self.ant_sig_present, self.AntennaSignal), - ('ant_noise', self.ant_noise_present, self.AntennaNoise), - ('lock_qual', self.lock_qual_present, self.LockQuality), - ('tx_attn', self.tx_attn_present, self.TxAttenuation), - ('db_tx_attn', self.db_tx_attn_present, self.DbTxAttenuation), - ('dbm_tx_power', self.dbm_tx_power_present, self.DbmTxPower), - ('ant', self.ant_present, self.Antenna), - ('db_ant_sig', self.db_ant_sig_present, self.DbAntennaSignal), - ('db_ant_noise', self.db_ant_noise_present, self.DbAntennaNoise), - ('rx_flags', self.rx_flags_present, self.RxFlags) - ] - for name, present_bit, parser in field_decoder: - if present_bit: - field = parser(buf) - field.data = '' - setattr(self, name, field) - self.fields.append(field) - buf = buf[len(field):] - - if len(self.data) > 0: - if self.flags_present and self.flags.fcs: - self.data = ieee80211.IEEE80211(self.data, fcs=self.flags.fcs) - else: - self.data = ieee80211.IEEE80211(self.data) - - class Antenna(dpkt.Packet): - __hdr__ = ( - ('index', 'B', 0), - ) - - class AntennaNoise(dpkt.Packet): - __hdr__ = ( - ('db', 'B', 0), - ) - - class AntennaSignal(dpkt.Packet): - __hdr__ = ( - ('db', 'B', 0), - ) - - class Channel(dpkt.Packet): - __hdr__ = ( - ('freq', 'H', 0), - ('flags', 'H', 0), - ) - - class FHSS(dpkt.Packet): - __hdr__ = ( - ('set', 'B', 0), - ('pattern', 'B', 0), - ) - - class Flags(dpkt.Packet): - __hdr__ = ( - ('val', 'B', 0), - ) - - @property - def fcs(self): return (self.val & _FCS_MASK) >> _FCS_SHIFT - - # TODO statement seems to have no effect - @fcs.setter - def fcs(self, v): (v << _FCS_SHIFT) | (self.val & ~_FCS_MASK) - - # Deprecated methods, will be removed in the future - # ================================================= - @deprecated - def _get_fcs_present(self): return self.fcs - - @deprecated - def _set_fcs_present(self, v): self.fcs = v - - # ================================================= - - class LockQuality(dpkt.Packet): - __hdr__ = ( - ('val', 'H', 0), - ) - - class RxFlags(dpkt.Packet): - __hdr__ = ( - ('val', 'H', 0), - ) - - class Rate(dpkt.Packet): - __hdr__ = ( - ('val', 'B', 0), - ) - - class TSFT(dpkt.Packet): - __hdr__ = ( - ('usecs', 'Q', 0), - ) - - class TxAttenuation(dpkt.Packet): - __hdr__ = ( - ('val', 'H', 0), - ) - - class DbTxAttenuation(dpkt.Packet): - __hdr__ = ( - ('db', 'H', 0), - ) - - class DbAntennaNoise(dpkt.Packet): - __hdr__ = ( - ('db', 'B', 0), - ) - - class DbAntennaSignal(dpkt.Packet): - __hdr__ = ( - ('db', 'B', 0), - ) - - class DbmTxPower(dpkt.Packet): - __hdr__ = ( - ('dbm', 'B', 0), - ) - - -def test_Radiotap(): - s = '\x00\x00\x00\x18\x6e\x48\x00\x00\x00\x02\x6c\x09\xa0\x00\xa8\x81\x02\x00\x00\x00\x00\x00\x00\x00' - rad = Radiotap(s) - assert(rad.version == 0) - assert(rad.present_flags == 0x6e480000) - assert(rad.tsft_present == 0) - assert(rad.flags_present == 1) - assert(rad.rate_present == 1) - assert(rad.channel_present == 1) - assert(rad.fhss_present == 0) - assert(rad.ant_sig_present == 1) - assert(rad.ant_noise_present == 1) - assert(rad.lock_qual_present == 0) - assert(rad.db_tx_attn_present == 0) - assert(rad.dbm_tx_power_present == 0) - assert(rad.ant_present == 1) - assert(rad.db_ant_sig_present == 0) - assert(rad.db_ant_noise_present == 0) - assert(rad.rx_flags_present == 1) - assert(rad.channel.freq == 0x6c09) - assert(rad.channel.flags == 0xa000) - assert(len(rad.fields) == 7) - - -def test_fcs(): - s = '\x00\x00\x1a\x00\x2f\x48\x00\x00\x34\x8f\x71\x09\x00\x00\x00\x00\x10\x0c\x85\x09\xc0\x00\xcc\x01\x00\x00' - rt = Radiotap(s) - assert(rt.flags_present == 1) - assert(rt.flags.fcs == 1) - - -if __name__ == '__main__': - test_Radiotap() - test_fcs() - print 'Tests Successful...'
\ No newline at end of file |