def setUp(self):
# Radio Tap(Flags,Rate,Channel,Antenna,DBMAntSignal,_
# FCSinHeader)+802.11 Data Frame+LLC SNAP+ARP Reply
self.frame_orig_1='\x00\x00\x18\x00\x0e\x58\x00\x00\x10\x6c\x6c\x09\x80\x04\x00\x1e\x00\x00\x00\x00\x00\x00\x00\x00\x08\x02\x2c\x00\x00\x1f\xe1\x19\xe4\xe4\x00\x1b\x9e\xce\x54\x09\x00\x1b\x9e\xce\x54\x09\xe0\xac\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x02\x00\x1b\x9e\xce\x54\x09\xc0\xa8\x01\x01\x00\x1f\xe1\x19\xe4\xe4\xc0\xa8\x01\x70\x01\x70\xe0\x00\x00\xfb\x94\x04\x00\x00\x16\x00\x00\x00\xe0\x00\x00\xfb\x17\x5c\xa6\xca'
self.rt1 = RadioTap(self.frame_orig_1)
# RadioTap(TSTF,Flags,Rate,DBMAntSignal,DBMAntNoise,_
# Antenna,XChannel)+802.11 Data Frame+LLC SNAP+ARP Request
self.frame_orig_2='\x00\x00\x20\x00\x67\x08\x04\x00\x30\x03\x1a\x25\x00\x00\x00\x00\x22\x0c\xd9\xa0\x02\x00\x00\x00\x40\x01\x00\x00\x3c\x14\x24\x11\x08\x02\x00\x00\xff\xff\xff\xff\xff\xff\x06\x03\x7f\x07\xa0\x16\x00\x19\xe3\xd3\x53\x52\x90\x7f\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x01\x00\x19\xe3\xd3\x53\x52\xa9\xfe\xf7\x00\x00\x00\x00\x00\x00\x00\x43\x08\x0e\x36'
self.rt2 = RadioTap(self.frame_orig_2)
def test_29_radiotap_length_field(self):
"""Test RadioTap header length field"""
# RadioTap from scratch calling get_length() and then get_packet()
rt = RadioTap()
# 0x08 bytes is the minimal headers size:
# 1 byte Revision
# 1 byte pad
# 2 bytes header length
# 4 bytes present flags
self.assertEqual(rt.get_header_length(), 0x08)
raw_packet = rt.get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00")
# RadioTap from scratch without call to get_length()
raw_packet = RadioTap().get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00")
def test_29_radiotap_length_field(self):
"""Test RadioTap header length field"""
# RadioTap from scratch calling get_length() and then get_packet()
rt = RadioTap()
# 0x08 bytes is the minimal headers size:
# 1 byte Revision
# 1 byte pad
# 2 bytes header length
# 4 bytes present flags
self.assertEqual(rt.get_header_length(), 0x08)
raw_packet = rt.get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00")
# RadioTap from scratch without call to get_length()
raw_packet = RadioTap().get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00")
def test_30_radiotap_length_filed_with_payload(self):
"""Test RadioTap header length field with payload"""
# RadioTap from scratch calling get_length() and then get_packet()
rt = RadioTap()
self.assertEqual(rt.get_header_length(), 0x08)
data = Data("aa")
rt.contains(data)
self.assertEqual(rt.get_header_length(),
0x08) # The header length is the same
raw_packet = rt.get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00aa")
def test_30_radiotap_length_filed_with_payload(self):
"""Test RadioTap header length field with payload"""
# RadioTap from scratch calling get_length() and then get_packet()
rt = RadioTap()
self.assertEqual(rt.get_header_length(), 0x08)
data = Data("aa")
rt.contains(data)
self.assertEqual(rt.get_header_length(), 0x08) # The header length is the same
raw_packet = rt.get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00aa")
class TestRadioTap(unittest.TestCase):
def setUp(self):
# Radio Tap(Flags,Rate,Channel,Antenna,DBMAntSignal,_
# FCSinHeader)+802.11 Data Frame+LLC SNAP+ARP Reply
self.frame_orig_1='\x00\x00\x18\x00\x0e\x58\x00\x00\x10\x6c\x6c\x09\x80\x04\x00\x1e\x00\x00\x00\x00\x00\x00\x00\x00\x08\x02\x2c\x00\x00\x1f\xe1\x19\xe4\xe4\x00\x1b\x9e\xce\x54\x09\x00\x1b\x9e\xce\x54\x09\xe0\xac\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x02\x00\x1b\x9e\xce\x54\x09\xc0\xa8\x01\x01\x00\x1f\xe1\x19\xe4\xe4\xc0\xa8\x01\x70\x01\x70\xe0\x00\x00\xfb\x94\x04\x00\x00\x16\x00\x00\x00\xe0\x00\x00\xfb\x17\x5c\xa6\xca'
self.rt1 = RadioTap(self.frame_orig_1)
# RadioTap(TSTF,Flags,Rate,DBMAntSignal,DBMAntNoise,_
# Antenna,XChannel)+802.11 Data Frame+LLC SNAP+ARP Request
self.frame_orig_2='\x00\x00\x20\x00\x67\x08\x04\x00\x30\x03\x1a\x25\x00\x00\x00\x00\x22\x0c\xd9\xa0\x02\x00\x00\x00\x40\x01\x00\x00\x3c\x14\x24\x11\x08\x02\x00\x00\xff\xff\xff\xff\xff\xff\x06\x03\x7f\x07\xa0\x16\x00\x19\xe3\xd3\x53\x52\x90\x7f\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x01\x00\x19\xe3\xd3\x53\x52\xa9\xfe\xf7\x00\x00\x00\x00\x00\x00\x00\x43\x08\x0e\x36'
self.rt2 = RadioTap(self.frame_orig_2)
def test_01_sizes(self):
'Test RadioTap frame sizes'
self.assertEqual(self.rt1.get_size(), len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(), 24)
self.assertEqual(self.rt1.get_body_size(), len(self.frame_orig_1)-24)
self.assertEqual(self.rt1.get_tail_size(), 0)
self.assertEqual(self.rt2.get_size(), len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(), 32)
self.assertEqual(self.rt2.get_body_size(), len(self.frame_orig_2)-32)
self.assertEqual(self.rt2.get_tail_size(), 0)
def test_02_version(self):
'Test RadioTap version getter/setter'
self.assertEqual(self.rt1.get_version(), 0x00)
self.rt1.set_version(1)
self.assertEqual(self.rt1.get_version(), 0x01)
self.assertEqual(self.rt2.get_version(), 0x00)
self.rt2.set_version(1)
self.assertEqual(self.rt2.get_version(), 0x01)
def test_03_present(self):
'Test RadioTap present getter'
self.assertEqual(self.rt1.get_present(), 0x0000580e)
self.assertEqual(self.rt2.get_present(), 0x00040867)
def test_04_present_bits(self):
'Test RadioTap present bits tester'
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TSFT), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_FLAGS), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_RATE), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_CHANNEL), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_FHSS), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DBM_ANTSIGNAL), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DBM_ANTNOISE), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_LOCK_QUALITY), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TX_ATTENUATION), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DB_TX_ATTENUATION), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DBM_TX_POWER), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_ANTENNA), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DB_ANTSIGNAL), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DB_ANTNOISE), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_FCS_IN_HEADER), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TX_FLAGS), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_RTS_RETRIES), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DATA_RETRIES), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_XCHANNEL), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_EXT), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_TSFT), True)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_FLAGS), True)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_RATE), True)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_CHANNEL), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_FHSS), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_DBM_ANTSIGNAL), True)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_DBM_ANTNOISE), True)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_LOCK_QUALITY), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_TX_ATTENUATION), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_DB_TX_ATTENUATION), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_DBM_TX_POWER), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_ANTENNA), True)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_DB_ANTSIGNAL), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_DB_ANTNOISE), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_FCS_IN_HEADER), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_TX_FLAGS), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_RTS_RETRIES), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_DATA_RETRIES), False)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_XCHANNEL), True)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_EXT), False)
def test_05_tsft(self):
'Test RadioTap tstf getter'
self.assertEqual(self.rt1.get_tsft(), None)
self.assertEqual(self.rt2.get_tsft(), 622461744)
def test_06_tsft(self):
'Test RadioTap tstf getter/setter'
# When the field is new
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.rt1.set_tsft(0x0102030405060708)
self.assertEqual(self.rt1.get_tsft(),0x0102030405060708)
self.assertEqual(self.rt1.get_header_size(),24+8)
# When exist the field
self.rt1.set_tsft(0x0807060504030201)
self.assertEqual(self.rt1.get_tsft(),0x0807060504030201)
self.assertEqual(self.rt1.get_header_size(),24+8)
def test_07_unset_fields(self):
'Test RadioTap unset field'
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_FLAGS), True)
self.rt1.unset_field(RadioTap.RTF_FLAGS)
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1)-1)
self.assertEqual(self.rt1.get_header_size(),24-1)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_FLAGS), False)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_TSFT), True)
self.rt2.unset_field(RadioTap.RTF_TSFT)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)-8)
self.assertEqual(self.rt2.get_header_size(),32-8)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_TSFT), False)
def test_08_flags_field(self):
'Test RadioTap flags getter/setter'
# When exist the field
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt1.get_flags(),0x10)
self.rt1.set_flags(0xAB)
self.assertEqual(self.rt1.get_flags(),0xAB)
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_flags(),0x22)
self.rt2.set_flags(0xAB)
self.assertEqual(self.rt2.get_flags(),0xAB)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
# TODO: Test the size when the field is new
def test_09_rate_field(self):
'Test RadioTap rate getter/setter'
# When exist the field
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt1.get_rate(),0x6c)
self.rt1.set_rate(0xAB)
self.assertEqual(self.rt1.get_rate(),0xAB)
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_rate(),0x0c)
self.rt2.set_rate(0xAB)
self.assertEqual(self.rt2.get_rate(),0xAB)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
# TODO: Test the size when the field is new
def test_10_channel_field(self):
'Test RadioTap channel getter/setter'
# When exist the field
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt1.get_channel(),(2412,0x0480))
self.rt1.set_channel( freq=1234, flags=0x5678 )
self.assertEqual(self.rt1.get_channel(),(1234,0x5678))
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_channel(),None)
self.rt2.set_channel( freq=1234, flags=0x5678 )
self.assertEqual(self.rt2.get_channel(),(1234,0x5678))
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+4)
self.assertEqual(self.rt2.get_header_size(),32+4)
def test_11_fhss_field(self):
'Test RadioTap FHSS getter/setter'
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_FHSS(),None)
self.rt2.set_FHSS( hop_set=0xAB, hop_pattern=0xCD )
self.assertEqual(self.rt2.get_FHSS(),(0xAB,0xCD))
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+2)
self.assertEqual(self.rt2.get_header_size(),32+2)
def test_12_dBm_ant_signal_field(self):
'Test RadioTap dBm Antenna Signal getter/setter'
# When exist the field
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_dBm_ant_signal(),0xd9)
self.rt2.set_dBm_ant_signal( signal=0xF1 )
self.assertEqual(self.rt2.get_dBm_ant_signal(),0xF1)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
# When the field is new
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt1.get_dBm_ant_signal(),None)
self.rt1.set_dBm_ant_signal( signal=0xF1 )
self.assertEqual(self.rt1.get_dBm_ant_signal(),0xF1)
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1)+1)
self.assertEqual(self.rt1.get_header_size(),24+1)
def test_13_dBm_ant_noise_field(self):
'Test RadioTap dBm Antenna Noise getter/setter'
# When exist the field
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_dBm_ant_noise(),0xa0)
self.rt2.set_dBm_ant_noise( signal=0xF1 )
self.assertEqual(self.rt2.get_dBm_ant_noise(),0xF1)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
# When the field is new
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt1.get_dBm_ant_noise(),None)
self.rt1.set_dBm_ant_noise( signal=0xF1 )
self.assertEqual(self.rt1.get_dBm_ant_noise(),0xF1)
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1)+1)
self.assertEqual(self.rt1.get_header_size(),24+1)
def test_14_lock_quality_field(self):
'Test RadioTap Lock Quality getter/setter'
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_lock_quality(),None)
self.rt2.set_lock_quality(quality=0xABBA )
self.assertEqual(self.rt2.get_lock_quality(),0xABBA)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+2)
self.assertEqual(self.rt2.get_header_size(),32+2)
def test_15_tx_attenuation_field(self):
'Test RadioTap Tx Attenuation getter/setter'
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_tx_attenuation(),None)
self.rt2.set_tx_attenuation(power=0xABBA )
self.assertEqual(self.rt2.get_tx_attenuation(),0xABBA)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+2)
self.assertEqual(self.rt2.get_header_size(),32+2)
def test_16_dB_tx_attenuation_field(self):
'Test RadioTap dB Tx Attenuation getter/setter'
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_dB_tx_attenuation(),None)
self.rt2.set_dB_tx_attenuation(power=0xABBA )
self.assertEqual(self.rt2.get_dB_tx_attenuation(),0xABBA)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+2)
self.assertEqual(self.rt2.get_header_size(),32+2)
def test_17_dBm_tx_power_field(self):
'Test RadioTap dBm Tx Power getter/setter'
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_dBm_tx_power(),None)
self.rt2.set_dBm_tx_power(power=-8)
self.assertEqual(self.rt2.get_dBm_tx_power(),-8)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+1)
self.assertEqual(self.rt2.get_header_size(),32+1)
def test_18_antenna_field(self):
'Test RadioTap Antenna getter/setter'
# When exist the field
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_antenna(),0x02)
self.rt2.set_antenna( antenna_index=0xF1 )
self.assertEqual(self.rt2.get_antenna(),0xF1)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt1.get_antenna(),0x00)
self.rt1.set_antenna( antenna_index=0xF1 )
self.assertEqual(self.rt1.get_antenna(),0xF1)
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
# TODO: When the field is new
def test_19_dB_ant_signal_field(self):
'Test RadioTap dB Antenna Signal getter/setter'
# When exist the field
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt1.get_dB_ant_signal(),0x1e)
self.rt1.set_dB_ant_signal( signal=0xF1 )
self.assertEqual(self.rt1.get_dB_ant_signal(),0xF1)
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_dB_ant_signal(),None)
self.rt2.set_dB_ant_signal( signal=0xF1 )
self.assertEqual(self.rt2.get_dB_ant_signal(),0xF1)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+1)
self.assertEqual(self.rt2.get_header_size(),32+1)
def test_20_dB_ant_noise_field(self):
'Test RadioTap dB Antenna Noise getter/setter'
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_dB_ant_noise(),None)
self.rt2.set_dB_ant_noise( signal=0xF1 )
self.assertEqual(self.rt2.get_dB_ant_noise(),0xF1)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+1)
self.assertEqual(self.rt2.get_header_size(),32+1)
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt1.get_dB_ant_noise(),None)
self.rt1.set_dB_ant_noise( signal=0xF1 )
self.assertEqual(self.rt1.get_dB_ant_noise(),0xF1)
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1)+1)
self.assertEqual(self.rt1.get_header_size(),24+1)
# TODO: When exist the field
## def test_21_rx_flags_field(self):
## 'Test RadioTap RX Flags getter/setter'
##
## # When the field is new
## self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
## self.assertEqual(self.rt2.get_header_size(),32)
## self.assertEqual(self.rt2.get_rx_flags(),None)
## self.rt2.set_rx_flags( signal=0xABBA )
## self.assertEqual(self.rt2.get_rx_flags(),0xABBA)
## self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+2)
## self.assertEqual(self.rt2.get_header_size(),32+2)
##
## # TODO: When exist the field
def test_22_FCS_in_header_field(self):
'Test RadioTap FCS in header getter/setter'
# When exist the field
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
self.assertEqual(self.rt1.get_FCS_in_header(),0x00000000)
self.rt1.set_FCS_in_header( fcs=0x89ABCDEF )
self.assertEqual(self.rt1.get_FCS_in_header(),0x89ABCDEF)
self.assertEqual(self.rt1.get_size(),len(self.frame_orig_1))
self.assertEqual(self.rt1.get_header_size(),24)
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_FCS_in_header(),None)
self.rt2.set_FCS_in_header( fcs=0x89ABCDEF )
self.assertEqual(self.rt2.get_FCS_in_header(),0x89ABCDEF)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+4)
self.assertEqual(self.rt2.get_header_size(),32+4)
## def test_23_rssi_field(self):
## 'Test RadioTap RSSI getter/setter'
##
## # When the field is new
## self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
## self.assertEqual(self.rt2.get_header_size(),32)
## self.assertEqual(self.rt2.get_RSSI(),None)
## self.rt2.set_RSSI( rssi=0xBA, max_rssi=0xAB )
## self.assertEqual(self.rt2.get_RSSI(),( 0xBA, 0xAB))
## self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+2)
## self.assertEqual(self.rt2.get_header_size(),32+2)
##
## # TODO: When exist the field
def test_24_RTS_retries_field(self):
'Test RadioTap RTS retries getter/setter'
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_RTS_retries(),None)
self.rt2.set_RTS_retries( retries=0xBA )
self.assertEqual(self.rt2.get_RTS_retries(), 0xBA)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+1)
self.assertEqual(self.rt2.get_header_size(),32+1)
# TODO: When exist the field
def test_25_tx_flags_field(self):
'Test RadioTap TX flags getter/setter'
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_tx_flags(),None)
self.rt2.set_tx_flags( flags=0xABBA )
self.assertEqual(self.rt2.get_tx_flags(),0xABBA)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+2)
self.assertEqual(self.rt2.get_header_size(),32+2)
# TODO: When exist the field
def test_26_xchannel_field(self):
'Test RadioTap xchannel getter/setter'
(ch_type,ch_freq,ch_num,ch_maxpower)=self.rt2.get_xchannel()
self.assertEqual(ch_type,0x00000140)
self.assertEqual(ch_freq,5180)
self.assertEqual(ch_num,36)
self.assertEqual(ch_maxpower,0x11)
(ch_type,ch_freq,ch_num,ch_maxpower)=(0x12345678, 1234, 12, 34)
self.rt2.set_xchannel(flags=ch_type, freq=ch_freq, channel=ch_num, maxpower=ch_maxpower)
(nch_type,nch_freq,nch_num,nch_maxpower)=self.rt2.get_xchannel()
self.assertEqual(ch_type,nch_type)
self.assertEqual(ch_freq,nch_freq)
self.assertEqual(ch_num,nch_num)
self.assertEqual(ch_maxpower,nch_maxpower)
def test_27_data_retries_field(self):
'Test RadioTap Data retries getter/setter'
# When the field is new
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
self.assertEqual(self.rt2.get_header_size(),32)
self.assertEqual(self.rt2.get_data_retries(),None)
self.rt2.set_data_retries( retries=0xAB )
self.assertEqual(self.rt2.get_data_retries(),0xAB)
self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+1)
self.assertEqual(self.rt2.get_header_size(),32+1)
# TODO: When exist the field
## def test_28_hardware_queue_field(self):
## 'Test RadioTap Hardware Queue getter/setter'
##
## # When the field is new
## self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
## self.assertEqual(self.rt2.get_header_size(),32)
## self.assertEqual(self.rt2.get_hardware_queue(),None)
## self.rt2.set_hardware_queue( queue=0xAB )
## self.assertEqual(self.rt2.get_hardware_queue(),0xAB)
## self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+1)
## self.assertEqual(self.rt2.get_header_size(),32+1)
##
## # TODO: When exist the field
def test_29_radiotap_length_field(self):
'Test RadioTap header length field'
# RadioTap from scratch calling get_length() and then get_packet()
rt = RadioTap()
# 0x08 bytes is the minimal headers size:
# 1 byte Revision
# 1 byte pad
# 2 bytes header length
# 4 bytes present flags
self.assertEqual(rt.get_header_length(), 0x08)
raw_packet = rt.get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00")
# RadioTap from scratch without call to get_length()
raw_packet = RadioTap().get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00")
def test_30_radiotap_length_filed_with_payload(self):
'Test RadioTap header length field with payload'
# RadioTap from scratch calling get_length() and then get_packet()
rt = RadioTap()
self.assertEqual(rt.get_header_length(), 0x08)
data = Data("aa")
rt.contains(data)
self.assertEqual(rt.get_header_length(), 0x08) # The header length is the same
raw_packet = rt.get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00aa")
def setUp(self):
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Reply
self.frame_0 = str()
self.frame_0 += '\x00' # Header Revision
self.frame_0 += '\x00' # Header Pad
self.frame_0 += '\x18\x00' # Header Length
self.frame_0 += '\x0e\x58\x00\x00' # Present Flags (flags, rate, channel, antenna, db antenna signal, rx flags)
self.frame_0 += '\x10' # Flags (FCS at end)
self.frame_0 += '\x6c' # Data Rate
self.frame_0 += '\x6c\x09' # Channel Frequency
self.frame_0 += '\x80\x04' # Channel Type
self.frame_0 += '\x00' # Antenna
self.frame_0 += '\x1e' # SSI Signal
self.frame_0 += '\x00\x00' # RX Flags
self.frame_0 += '\x00\x00\x00\x00\x00\x00\x08\x02\x2c\x00\x00\x1f\xe1\x19\xe4\xe4\x00\x1b\x9e\xce\x54\x09\x00' \
'\x1b\x9e\xce\x54\x09\xe0\xac\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x02' \
'\x00\x1b\x9e\xce\x54\x09\xc0\xa8\x01\x01\x00\x1f\xe1\x19\xe4\xe4\xc0\xa8\x01\x70\x01\x70\xe0' \
'\x00\x00\xfb\x94\x04\x00\x00\x16\x00\x00\x00\xe0\x00\x00\xfb\x17\x5c\xa6\xca'
self.rt0 = RadioTap(self.frame_0)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_1 = str()
self.frame_1 += '\x00' # Header Revision
self.frame_1 += '\x00' # Header Pad
self.frame_1 += '\x20\x00' # Header Length
# Present Flags (tsft, flags, rate, dbm antenna signal, dbm antenna noise, antenna, channel+)
self.frame_1 += '\x67\x08\x04\x00'
self.frame_1 += '\x30\x03\x1a\x25\x00\x00\x00\x00' # TSFT
self.frame_1 += '\x22' # Flags (Data pad, preamble short)
self.frame_1 += '\x0c' # Data Rate
self.frame_1 += '\xd9' # SSI Signal
self.frame_1 += '\xa0' # SSI Noise
self.frame_1 += '\x02' # Antenna
self.frame_1 += '\x00\x00\x00' # ???
self.frame_1 += '\x40\x01\x00\x00' # Channel Type
self.frame_1 += '\x3c\x14' # Channel Frequency
self.frame_1 += '\x24' # Channel Number
self.frame_1 += '\x11' # ???
self.frame_1 += '\x08\x02\x00\x00\xff\xff\xff\xff\xff\xff\x06\x03\x7f\x07\xa0\x16\x00\x19\xe3\xd3\x53\x52' \
'\x90\x7f\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x01\x00\x19\xe3\xd3' \
'\x53\x52\xa9\xfe\xf7\x00\x00\x00\x00\x00\x00\x00\x43\x08\x0e\x36'
self.rt1 = RadioTap(self.frame_1)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_2 = str()
self.frame_2 += '\x00' # Header Revision
self.frame_2 += '\x00' # Header Pad
self.frame_2 += '\x24\x00' # Header Length
# Present Flags (tsft, flags, rate, channel, dbm antenna signal, rx flags, ns next, ext)
self.frame_2 += '\x2f\x40\x00\xa0'
self.frame_2 += '\x20\x08\x00\x00' # Present Flags (dbm antenna signal, antenna)
self.frame_2 += '\x00\x00\x00\x00' # ???
self.frame_2 += '\x97\xd3\x26\x44\x06\x00\x00\x00' # TSFT
self.frame_2 += '\x10' # Flags
self.frame_2 += '\x02' # Data Rate
self.frame_2 += '\x6c\x09' # Channel Frequency
self.frame_2 += '\xc0\x00' # Channel Type
self.frame_2 += '\xa6' # SSI Signal
self.frame_2 += '\x00' # ???
self.frame_2 += '\x00\x00' # RX flags
self.frame_2 += '\xa6' # SSI Signal
self.frame_2 += '\x00' # ???
self.frame_2 += '\xd4\x00\x00\x00\x9c\x04\xeb\x4d\xdb\x53\x8d\xf3\xc6\xc3'
self.rt2 = RadioTap(self.frame_2)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_3 = str()
self.frame_3 += '\x00' # Header Revision
self.frame_3 += '\x00' # Header Pad
self.frame_3 += '\x24\x00' # Header Length
# Present Flags (tsft, flags, rate, channel, dbm antenna signal, rx flags, ns next, ext)
self.frame_3 += '\x2f\x40\x00\xa0'
self.frame_3 += '\x20\x08\x00\x80' # Present Flags (dbm antenna signal, antenna)
self.frame_3 += '\x00\x00\x00\x00' # Present Flags
# self.frame_3 += '\x00\x00\x00\x00' # ???
self.frame_3 += '\x97\xd3\x26\x44\x06\x00\x00\x00' # TSFT
self.frame_3 += '\x10' # Flags
self.frame_3 += '\x02' # Data Rate
self.frame_3 += '\x6c\x09' # Channel Frequency
self.frame_3 += '\xc0\x00' # Channel Type
self.frame_3 += '\xa6' # SSI Signal
self.frame_3 += '\x00' # ???
self.frame_3 += '\x00\x00' # RX flags
self.frame_3 += '\xa6' # SSI Signal
self.frame_3 += '\x00' # ???
self.frame_3 += '\xd4\x00\x00\x00\x9c\x04\xeb\x4d\xdb\x53\x8d\xf3\xc6\xc3'
self.rt3 = RadioTap(self.frame_3)
class TestRadioTap(unittest.TestCase):
def setUp(self):
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Reply
self.frame_0 = str()
self.frame_0 += '\x00' # Header Revision
self.frame_0 += '\x00' # Header Pad
self.frame_0 += '\x18\x00' # Header Length
self.frame_0 += '\x0e\x58\x00\x00' # Present Flags (flags, rate, channel, antenna, db antenna signal, rx flags)
self.frame_0 += '\x10' # Flags (FCS at end)
self.frame_0 += '\x6c' # Data Rate
self.frame_0 += '\x6c\x09' # Channel Frequency
self.frame_0 += '\x80\x04' # Channel Type
self.frame_0 += '\x00' # Antenna
self.frame_0 += '\x1e' # SSI Signal
self.frame_0 += '\x00\x00' # RX Flags
self.frame_0 += '\x00\x00\x00\x00\x00\x00\x08\x02\x2c\x00\x00\x1f\xe1\x19\xe4\xe4\x00\x1b\x9e\xce\x54\x09\x00' \
'\x1b\x9e\xce\x54\x09\xe0\xac\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x02' \
'\x00\x1b\x9e\xce\x54\x09\xc0\xa8\x01\x01\x00\x1f\xe1\x19\xe4\xe4\xc0\xa8\x01\x70\x01\x70\xe0' \
'\x00\x00\xfb\x94\x04\x00\x00\x16\x00\x00\x00\xe0\x00\x00\xfb\x17\x5c\xa6\xca'
self.rt0 = RadioTap(self.frame_0)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_1 = str()
self.frame_1 += '\x00' # Header Revision
self.frame_1 += '\x00' # Header Pad
self.frame_1 += '\x20\x00' # Header Length
# Present Flags (tsft, flags, rate, dbm antenna signal, dbm antenna noise, antenna, channel+)
self.frame_1 += '\x67\x08\x04\x00'
self.frame_1 += '\x30\x03\x1a\x25\x00\x00\x00\x00' # TSFT
self.frame_1 += '\x22' # Flags (Data pad, preamble short)
self.frame_1 += '\x0c' # Data Rate
self.frame_1 += '\xd9' # SSI Signal
self.frame_1 += '\xa0' # SSI Noise
self.frame_1 += '\x02' # Antenna
self.frame_1 += '\x00\x00\x00' # ???
self.frame_1 += '\x40\x01\x00\x00' # Channel Type
self.frame_1 += '\x3c\x14' # Channel Frequency
self.frame_1 += '\x24' # Channel Number
self.frame_1 += '\x11' # ???
self.frame_1 += '\x08\x02\x00\x00\xff\xff\xff\xff\xff\xff\x06\x03\x7f\x07\xa0\x16\x00\x19\xe3\xd3\x53\x52' \
'\x90\x7f\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x01\x00\x19\xe3\xd3' \
'\x53\x52\xa9\xfe\xf7\x00\x00\x00\x00\x00\x00\x00\x43\x08\x0e\x36'
self.rt1 = RadioTap(self.frame_1)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_2 = str()
self.frame_2 += '\x00' # Header Revision
self.frame_2 += '\x00' # Header Pad
self.frame_2 += '\x24\x00' # Header Length
# Present Flags (tsft, flags, rate, channel, dbm antenna signal, rx flags, ns next, ext)
self.frame_2 += '\x2f\x40\x00\xa0'
self.frame_2 += '\x20\x08\x00\x00' # Present Flags (dbm antenna signal, antenna)
self.frame_2 += '\x00\x00\x00\x00' # ???
self.frame_2 += '\x97\xd3\x26\x44\x06\x00\x00\x00' # TSFT
self.frame_2 += '\x10' # Flags
self.frame_2 += '\x02' # Data Rate
self.frame_2 += '\x6c\x09' # Channel Frequency
self.frame_2 += '\xc0\x00' # Channel Type
self.frame_2 += '\xa6' # SSI Signal
self.frame_2 += '\x00' # ???
self.frame_2 += '\x00\x00' # RX flags
self.frame_2 += '\xa6' # SSI Signal
self.frame_2 += '\x00' # ???
self.frame_2 += '\xd4\x00\x00\x00\x9c\x04\xeb\x4d\xdb\x53\x8d\xf3\xc6\xc3'
self.rt2 = RadioTap(self.frame_2)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_3 = str()
self.frame_3 += '\x00' # Header Revision
self.frame_3 += '\x00' # Header Pad
self.frame_3 += '\x24\x00' # Header Length
# Present Flags (tsft, flags, rate, channel, dbm antenna signal, rx flags, ns next, ext)
self.frame_3 += '\x2f\x40\x00\xa0'
self.frame_3 += '\x20\x08\x00\x80' # Present Flags (dbm antenna signal, antenna)
self.frame_3 += '\x00\x00\x00\x00' # Present Flags
# self.frame_3 += '\x00\x00\x00\x00' # ???
self.frame_3 += '\x97\xd3\x26\x44\x06\x00\x00\x00' # TSFT
self.frame_3 += '\x10' # Flags
self.frame_3 += '\x02' # Data Rate
self.frame_3 += '\x6c\x09' # Channel Frequency
self.frame_3 += '\xc0\x00' # Channel Type
self.frame_3 += '\xa6' # SSI Signal
self.frame_3 += '\x00' # ???
self.frame_3 += '\x00\x00' # RX flags
self.frame_3 += '\xa6' # SSI Signal
self.frame_3 += '\x00' # ???
self.frame_3 += '\xd4\x00\x00\x00\x9c\x04\xeb\x4d\xdb\x53\x8d\xf3\xc6\xc3'
self.rt3 = RadioTap(self.frame_3)
def test_01_sizes(self):
"""Test RadioTap frame sizes"""
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_body_size(), len(self.frame_0)-24)
self.assertEqual(self.rt0.get_tail_size(), 0)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_body_size(), len(self.frame_1)-32)
self.assertEqual(self.rt1.get_tail_size(), 0)
def test_02_version(self):
"""Test RadioTap version getter/setter"""
self.assertEqual(self.rt0.get_version(), 0x00)
self.rt0.set_version(1)
self.assertEqual(self.rt0.get_version(), 0x01)
self.assertEqual(self.rt1.get_version(), 0x00)
self.rt1.set_version(1)
self.assertEqual(self.rt1.get_version(), 0x01)
def test_03_present(self):
"""Test RadioTap present getter"""
self.assertEqual(self.rt0.get_present(), 0x0000580e)
self.assertEqual(self.rt1.get_present(), 0x00040867)
def test_04_present_bits(self):
"""Test RadioTap present bits tester"""
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_TSFT), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_FLAGS), True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_RATE), True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_CHANNEL), True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_FHSS), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DBM_ANTSIGNAL), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DBM_ANTNOISE), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_LOCK_QUALITY), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_TX_ATTENUATION), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DB_TX_ATTENUATION), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DBM_TX_POWER), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_ANTENNA), True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DB_ANTSIGNAL), True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DB_ANTNOISE), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_FCS_IN_HEADER), True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_TX_FLAGS), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_RTS_RETRIES), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DATA_RETRIES), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_XCHANNEL), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_EXT), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TSFT), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_FLAGS), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_RATE), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_CHANNEL), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_FHSS), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DBM_ANTSIGNAL), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DBM_ANTNOISE), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_LOCK_QUALITY), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TX_ATTENUATION), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DB_TX_ATTENUATION), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DBM_TX_POWER), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_ANTENNA), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DB_ANTSIGNAL), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DB_ANTNOISE), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_FCS_IN_HEADER), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TX_FLAGS), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_RTS_RETRIES), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DATA_RETRIES), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_XCHANNEL), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_EXT), False)
def test_05_tsft(self):
"""Test RadioTap tstf getter"""
self.assertEqual(self.rt0.get_tsft(), None)
self.assertEqual(self.rt1.get_tsft(), 622461744)
def test_06_tsft(self):
"""Test RadioTap tstf getter/setter"""
# When the field is new
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.rt0.set_tsft(0x0102030405060708)
self.assertEqual(self.rt0.get_tsft(), 0x0102030405060708)
self.assertEqual(self.rt0.get_header_size(), 24+8)
# When exist the field
self.rt0.set_tsft(0x0807060504030201)
self.assertEqual(self.rt0.get_tsft(), 0x0807060504030201)
self.assertEqual(self.rt0.get_header_size(), 24+8)
def test_07_unset_fields(self):
"""Test RadioTap unset field"""
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_FLAGS), True)
self.rt0.unset_field(RadioTap.RTF_FLAGS)
self.assertEqual(self.rt0.get_size(), len(self.frame_0)-1)
self.assertEqual(self.rt0.get_header_size(), 24-1)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_FLAGS), False)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TSFT), True)
self.rt1.unset_field(RadioTap.RTF_TSFT)
self.assertEqual(self.rt1.get_size(), len(self.frame_1)-8)
self.assertEqual(self.rt1.get_header_size(), 32-8)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TSFT), False)
def test_08_flags_field(self):
"""Test RadioTap flags getter/setter"""
# When exist the field
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_flags(), 0x10)
self.rt0.set_flags(0xAB)
self.assertEqual(self.rt0.get_flags(), 0xAB)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_flags(), 0x22)
self.rt1.set_flags(0xAB)
self.assertEqual(self.rt1.get_flags(), 0xAB)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
# TODO: Test the size when the field is new
def test_09_rate_field(self):
"""Test RadioTap rate getter/setter"""
# When exist the field
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_rate(), 0x6c)
self.rt0.set_rate(0xAB)
self.assertEqual(self.rt0.get_rate(), 0xAB)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_rate(), 0x0c)
self.rt1.set_rate(0xAB)
self.assertEqual(self.rt1.get_rate(), 0xAB)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
# TODO: Test the size when the field is new
def test_10_channel_field(self):
"""Test RadioTap channel getter/setter"""
# When exist the field
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_channel(), (2412, 0x0480))
self.rt0.set_channel(freq=1234, flags=0x5678)
self.assertEqual(self.rt0.get_channel(), (1234, 0x5678))
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_channel(), None)
self.rt1.set_channel(freq=1234, flags=0x5678)
self.assertEqual(self.rt1.get_channel(), (1234, 0x5678))
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+4)
self.assertEqual(self.rt1.get_header_size(), 32+4)
def test_11_fhss_field(self):
"""Test RadioTap FHSS getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_FHSS(), None)
self.rt1.set_FHSS(hop_set=0xAB, hop_pattern=0xCD)
self.assertEqual(self.rt1.get_FHSS(), (0xAB, 0xCD))
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+2)
self.assertEqual(self.rt1.get_header_size(), 32+2)
def test_12_dbm_ant_signal_field(self):
"""Test RadioTap dBm Antenna Signal getter/setter"""
# When exist the field
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dBm_ant_signal(), 0xd9)
self.rt1.set_dBm_ant_signal(signal=0xF1)
self.assertEqual(self.rt1.get_dBm_ant_signal(), 0xF1)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
# When the field is new
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_dBm_ant_signal(), None)
self.rt0.set_dBm_ant_signal(signal=0xF1)
self.assertEqual(self.rt0.get_dBm_ant_signal(), 0xF1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0)+1)
self.assertEqual(self.rt0.get_header_size(), 24+1)
def test_13_dbm_ant_noise_field(self):
"""Test RadioTap dBm Antenna Noise getter/setter"""
# When exist the field
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dBm_ant_noise(), 0xa0)
self.rt1.set_dBm_ant_noise(signal=0xF1)
self.assertEqual(self.rt1.get_dBm_ant_noise(), 0xF1)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
# When the field is new
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_dBm_ant_noise(), None)
self.rt0.set_dBm_ant_noise(signal=0xF1)
self.assertEqual(self.rt0.get_dBm_ant_noise(), 0xF1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0)+1)
self.assertEqual(self.rt0.get_header_size(), 24+1)
def test_14_lock_quality_field(self):
"""Test RadioTap Lock Quality getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_lock_quality(), None)
self.rt1.set_lock_quality(quality=0xABBA)
self.assertEqual(self.rt1.get_lock_quality(), 0xABBA)
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+2)
self.assertEqual(self.rt1.get_header_size(), 32+2)
def test_15_tx_attenuation_field(self):
"""Test RadioTap Tx Attenuation getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_tx_attenuation(), None)
self.rt1.set_tx_attenuation(power=0xABBA)
self.assertEqual(self.rt1.get_tx_attenuation(), 0xABBA)
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+2)
self.assertEqual(self.rt1.get_header_size(), 32+2)
def test_16_dB_tx_attenuation_field(self):
"""Test RadioTap dB Tx Attenuation getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dB_tx_attenuation(), None)
self.rt1.set_dB_tx_attenuation(power=0xABBA)
self.assertEqual(self.rt1.get_dB_tx_attenuation(), 0xABBA)
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+2)
self.assertEqual(self.rt1.get_header_size(), 32+2)
def test_17_dbm_tx_power_field(self):
"""Test RadioTap dBm Tx Power getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dBm_tx_power(), None)
self.rt1.set_dBm_tx_power(power=-8)
self.assertEqual(self.rt1.get_dBm_tx_power(), -8)
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+1)
self.assertEqual(self.rt1.get_header_size(), 32+1)
def test_18_antenna_field(self):
"""Test RadioTap Antenna getter/setter"""
# TODO: When the field is new
# When exist the field
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_antenna(), 0x02)
self.rt1.set_antenna(antenna_index=0xF1)
self.assertEqual(self.rt1.get_antenna(), 0xF1)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_antenna(), 0x00)
self.rt0.set_antenna(antenna_index=0xF1)
self.assertEqual(self.rt0.get_antenna(), 0xF1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
def test_19_db_ant_signal_field(self):
"""Test RadioTap dB Antenna Signal getter/setter"""
# When exist the field
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_dB_ant_signal(), 0x1e)
self.rt0.set_dB_ant_signal(signal=0xF1)
self.assertEqual(self.rt0.get_dB_ant_signal(), 0xF1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dB_ant_signal(), None)
self.rt1.set_dB_ant_signal(signal=0xF1)
self.assertEqual(self.rt1.get_dB_ant_signal(), 0xF1)
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+1)
self.assertEqual(self.rt1.get_header_size(), 32+1)
def test_20_db_ant_noise_field(self):
"""Test RadioTap dB Antenna Noise getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dB_ant_noise(), None)
self.rt1.set_dB_ant_noise(signal=0xF1)
self.assertEqual(self.rt1.get_dB_ant_noise(), 0xF1)
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+1)
self.assertEqual(self.rt1.get_header_size(), 32+1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_dB_ant_noise(), None)
self.rt0.set_dB_ant_noise(signal=0xF1)
self.assertEqual(self.rt0.get_dB_ant_noise(), 0xF1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0)+1)
self.assertEqual(self.rt0.get_header_size(), 24+1)
# def test_21_rx_flags_field(self):
# """Test RadioTap RX Flags getter/setter"""
# # When the field is new
# # TODO: When exist the field
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
# self.assertEqual(self.rt2.get_header_size(),32)
# self.assertEqual(self.rt2.get_rx_flags(),None)
# self.rt2.set_rx_flags( signal=0xABBA )
# self.assertEqual(self.rt2.get_rx_flags(),0xABBA)
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+2)
# self.assertEqual(self.rt2.get_header_size(),32+2)
def test_22_fcs_in_header_field(self):
"""Test RadioTap FCS in header getter/setter"""
# When exist the field
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_FCS_in_header(), 0x00000000)
self.rt0.set_FCS_in_header(fcs=0x89ABCDEF)
self.assertEqual(self.rt0.get_FCS_in_header(), 0x89ABCDEF)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_FCS_in_header(), None)
self.rt1.set_FCS_in_header(fcs=0x89ABCDEF)
self.assertEqual(self.rt1.get_FCS_in_header(), 0x89ABCDEF)
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+4)
self.assertEqual(self.rt1.get_header_size(), 32+4)
# def test_23_rssi_field(self):
# """Test RadioTap RSSI getter/setter"""
# # When the field is new
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
# self.assertEqual(self.rt2.get_header_size(),32)
# self.assertEqual(self.rt2.get_RSSI(),None)
# self.rt2.set_RSSI( rssi=0xBA, max_rssi=0xAB )
# self.assertEqual(self.rt2.get_RSSI(),( 0xBA, 0xAB))
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+2)
# self.assertEqual(self.rt2.get_header_size(),32+2)
# # TODO: When exist the field
def test_24_rts_retries_field(self):
"""Test RadioTap RTS retries getter/setter"""
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_RTS_retries(), None)
self.rt1.set_RTS_retries(retries=0xBA)
self.assertEqual(self.rt1.get_RTS_retries(), 0xBA)
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+1)
self.assertEqual(self.rt1.get_header_size(), 32+1)
# TODO: When exist the field
def test_25_tx_flags_field(self):
"""Test RadioTap TX flags getter/setter"""
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_tx_flags(), None)
self.rt1.set_tx_flags(flags=0xABBA)
self.assertEqual(self.rt1.get_tx_flags(), 0xABBA)
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+2)
self.assertEqual(self.rt1.get_header_size(), 32+2)
# TODO: When exist the field
def test_26_xchannel_field(self):
"""Test RadioTap xchannel getter/setter"""
(ch_type, ch_freq, ch_num, ch_maxpower) = self.rt1.get_xchannel()
self.assertEqual(ch_type, 0x00000140)
self.assertEqual(ch_freq, 5180)
self.assertEqual(ch_num, 36)
self.assertEqual(ch_maxpower, 0x11)
(ch_type, ch_freq, ch_num, ch_maxpower) = (0x12345678, 1234, 12, 34)
self.rt1.set_xchannel(flags=ch_type, freq=ch_freq, channel=ch_num, maxpower=ch_maxpower)
(nch_type, nch_freq, nch_num, nch_maxpower) = self.rt1.get_xchannel()
self.assertEqual(ch_type, nch_type)
self.assertEqual(ch_freq, nch_freq)
self.assertEqual(ch_num, nch_num)
self.assertEqual(ch_maxpower, nch_maxpower)
def test_27_data_retries_field(self):
"""Test RadioTap Data retries getter/setter"""
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_data_retries(), None)
self.rt1.set_data_retries(retries=0xAB)
self.assertEqual(self.rt1.get_data_retries(), 0xAB)
self.assertEqual(self.rt1.get_size(), len(self.frame_1)+1)
self.assertEqual(self.rt1.get_header_size(), 32+1)
# TODO: When exist the field
# def test_28_hardware_queue_field(self):
# """Test RadioTap Hardware Queue getter/setter"""
# # When the field is new
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
# self.assertEqual(self.rt2.get_header_size(),32)
# self.assertEqual(self.rt2.get_hardware_queue(),None)
# self.rt2.set_hardware_queue( queue=0xAB )
# self.assertEqual(self.rt2.get_hardware_queue(),0xAB)
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+1)
# self.assertEqual(self.rt2.get_header_size(),32+1)
# # TODO: When exist the field
def test_29_radiotap_length_field(self):
"""Test RadioTap header length field"""
# RadioTap from scratch calling get_length() and then get_packet()
rt = RadioTap()
# 0x08 bytes is the minimal headers size:
# 1 byte Revision
# 1 byte pad
# 2 bytes header length
# 4 bytes present flags
self.assertEqual(rt.get_header_length(), 0x08)
raw_packet = rt.get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00")
# RadioTap from scratch without call to get_length()
raw_packet = RadioTap().get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00")
def test_30_radiotap_length_filed_with_payload(self):
"""Test RadioTap header length field with payload"""
# RadioTap from scratch calling get_length() and then get_packet()
rt = RadioTap()
self.assertEqual(rt.get_header_length(), 0x08)
data = Data("aa")
rt.contains(data)
self.assertEqual(rt.get_header_length(), 0x08) # The header length is the same
raw_packet = rt.get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00aa")
def test_31_radiotap_present_flags_extended(self):
"""Test RadioTap extended present flags"""
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_EXT), True)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_RATE), True)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_CHANNEL), True)
self.assertEqual(self.rt2.get_channel(), (2412, 192))
self.assertEqual(self.rt2.get_rate(), 2)
self.assertEqual(self.rt2.get_dBm_ant_signal(), 0xa6)
self.assertEqual(self.rt3.get_present_bit(RadioTap.RTF_EXT), True)
self.assertEqual(self.rt3.get_present_bit(RadioTap.RTF_RATE), True)
self.assertEqual(self.rt3.get_present_bit(RadioTap.RTF_CHANNEL), True)
self.assertEqual(self.rt3.get_channel(), (2412, 192))
self.assertEqual(self.rt3.get_rate(), 2)
self.assertEqual(self.rt3.get_dBm_ant_signal(), 0xa6)
def setUp(self):
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Reply
self.frame_0 = str()
self.frame_0 += '\x00' # Header Revision
self.frame_0 += '\x00' # Header Pad
self.frame_0 += '\x18\x00' # Header Length
self.frame_0 += '\x0e\x58\x00\x00' # Present Flags (flags, rate, channel, antenna, db antenna signal, rx flags)
self.frame_0 += '\x10' # Flags (FCS at end)
self.frame_0 += '\x6c' # Data Rate
self.frame_0 += '\x6c\x09' # Channel Frequency
self.frame_0 += '\x80\x04' # Channel Type
self.frame_0 += '\x00' # Antenna
self.frame_0 += '\x1e' # SSI Signal
self.frame_0 += '\x00\x00' # RX Flags
self.frame_0 += '\x00\x00\x00\x00\x00\x00\x08\x02\x2c\x00\x00\x1f\xe1\x19\xe4\xe4\x00\x1b\x9e\xce\x54\x09\x00' \
'\x1b\x9e\xce\x54\x09\xe0\xac\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x02' \
'\x00\x1b\x9e\xce\x54\x09\xc0\xa8\x01\x01\x00\x1f\xe1\x19\xe4\xe4\xc0\xa8\x01\x70\x01\x70\xe0' \
'\x00\x00\xfb\x94\x04\x00\x00\x16\x00\x00\x00\xe0\x00\x00\xfb\x17\x5c\xa6\xca'
self.rt0 = RadioTap(self.frame_0)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_1 = str()
self.frame_1 += '\x00' # Header Revision
self.frame_1 += '\x00' # Header Pad
self.frame_1 += '\x20\x00' # Header Length
# Present Flags (tsft, flags, rate, dbm antenna signal, dbm antenna noise, antenna, channel+)
self.frame_1 += '\x67\x08\x04\x00'
self.frame_1 += '\x30\x03\x1a\x25\x00\x00\x00\x00' # TSFT
self.frame_1 += '\x22' # Flags (Data pad, preamble short)
self.frame_1 += '\x0c' # Data Rate
self.frame_1 += '\xd9' # SSI Signal
self.frame_1 += '\xa0' # SSI Noise
self.frame_1 += '\x02' # Antenna
self.frame_1 += '\x00\x00\x00' # ???
self.frame_1 += '\x40\x01\x00\x00' # Channel Type
self.frame_1 += '\x3c\x14' # Channel Frequency
self.frame_1 += '\x24' # Channel Number
self.frame_1 += '\x11' # ???
self.frame_1 += '\x08\x02\x00\x00\xff\xff\xff\xff\xff\xff\x06\x03\x7f\x07\xa0\x16\x00\x19\xe3\xd3\x53\x52' \
'\x90\x7f\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x01\x00\x19\xe3\xd3' \
'\x53\x52\xa9\xfe\xf7\x00\x00\x00\x00\x00\x00\x00\x43\x08\x0e\x36'
self.rt1 = RadioTap(self.frame_1)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_2 = str()
self.frame_2 += '\x00' # Header Revision
self.frame_2 += '\x00' # Header Pad
self.frame_2 += '\x24\x00' # Header Length
# Present Flags (tsft, flags, rate, channel, dbm antenna signal, rx flags, ns next, ext)
self.frame_2 += '\x2f\x40\x00\xa0'
self.frame_2 += '\x20\x08\x00\x00' # Present Flags (dbm antenna signal, antenna)
self.frame_2 += '\x00\x00\x00\x00' # ???
self.frame_2 += '\x97\xd3\x26\x44\x06\x00\x00\x00' # TSFT
self.frame_2 += '\x10' # Flags
self.frame_2 += '\x02' # Data Rate
self.frame_2 += '\x6c\x09' # Channel Frequency
self.frame_2 += '\xc0\x00' # Channel Type
self.frame_2 += '\xa6' # SSI Signal
self.frame_2 += '\x00' # ???
self.frame_2 += '\x00\x00' # RX flags
self.frame_2 += '\xa6' # SSI Signal
self.frame_2 += '\x00' # ???
self.frame_2 += '\xd4\x00\x00\x00\x9c\x04\xeb\x4d\xdb\x53\x8d\xf3\xc6\xc3'
self.rt2 = RadioTap(self.frame_2)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_3 = str()
self.frame_3 += '\x00' # Header Revision
self.frame_3 += '\x00' # Header Pad
self.frame_3 += '\x24\x00' # Header Length
# Present Flags (tsft, flags, rate, channel, dbm antenna signal, rx flags, ns next, ext)
self.frame_3 += '\x2f\x40\x00\xa0'
self.frame_3 += '\x20\x08\x00\x80' # Present Flags (dbm antenna signal, antenna)
self.frame_3 += '\x00\x00\x00\x00' # Present Flags
# self.frame_3 += '\x00\x00\x00\x00' # ???
self.frame_3 += '\x97\xd3\x26\x44\x06\x00\x00\x00' # TSFT
self.frame_3 += '\x10' # Flags
self.frame_3 += '\x02' # Data Rate
self.frame_3 += '\x6c\x09' # Channel Frequency
self.frame_3 += '\xc0\x00' # Channel Type
self.frame_3 += '\xa6' # SSI Signal
self.frame_3 += '\x00' # ???
self.frame_3 += '\x00\x00' # RX flags
self.frame_3 += '\xa6' # SSI Signal
self.frame_3 += '\x00' # ???
self.frame_3 += '\xd4\x00\x00\x00\x9c\x04\xeb\x4d\xdb\x53\x8d\xf3\xc6\xc3'
self.rt3 = RadioTap(self.frame_3)
class TestRadioTap(unittest.TestCase):
def setUp(self):
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Reply
self.frame_0 = str()
self.frame_0 += '\x00' # Header Revision
self.frame_0 += '\x00' # Header Pad
self.frame_0 += '\x18\x00' # Header Length
self.frame_0 += '\x0e\x58\x00\x00' # Present Flags (flags, rate, channel, antenna, db antenna signal, rx flags)
self.frame_0 += '\x10' # Flags (FCS at end)
self.frame_0 += '\x6c' # Data Rate
self.frame_0 += '\x6c\x09' # Channel Frequency
self.frame_0 += '\x80\x04' # Channel Type
self.frame_0 += '\x00' # Antenna
self.frame_0 += '\x1e' # SSI Signal
self.frame_0 += '\x00\x00' # RX Flags
self.frame_0 += '\x00\x00\x00\x00\x00\x00\x08\x02\x2c\x00\x00\x1f\xe1\x19\xe4\xe4\x00\x1b\x9e\xce\x54\x09\x00' \
'\x1b\x9e\xce\x54\x09\xe0\xac\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x02' \
'\x00\x1b\x9e\xce\x54\x09\xc0\xa8\x01\x01\x00\x1f\xe1\x19\xe4\xe4\xc0\xa8\x01\x70\x01\x70\xe0' \
'\x00\x00\xfb\x94\x04\x00\x00\x16\x00\x00\x00\xe0\x00\x00\xfb\x17\x5c\xa6\xca'
self.rt0 = RadioTap(self.frame_0)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_1 = str()
self.frame_1 += '\x00' # Header Revision
self.frame_1 += '\x00' # Header Pad
self.frame_1 += '\x20\x00' # Header Length
# Present Flags (tsft, flags, rate, dbm antenna signal, dbm antenna noise, antenna, channel+)
self.frame_1 += '\x67\x08\x04\x00'
self.frame_1 += '\x30\x03\x1a\x25\x00\x00\x00\x00' # TSFT
self.frame_1 += '\x22' # Flags (Data pad, preamble short)
self.frame_1 += '\x0c' # Data Rate
self.frame_1 += '\xd9' # SSI Signal
self.frame_1 += '\xa0' # SSI Noise
self.frame_1 += '\x02' # Antenna
self.frame_1 += '\x00\x00\x00' # ???
self.frame_1 += '\x40\x01\x00\x00' # Channel Type
self.frame_1 += '\x3c\x14' # Channel Frequency
self.frame_1 += '\x24' # Channel Number
self.frame_1 += '\x11' # ???
self.frame_1 += '\x08\x02\x00\x00\xff\xff\xff\xff\xff\xff\x06\x03\x7f\x07\xa0\x16\x00\x19\xe3\xd3\x53\x52' \
'\x90\x7f\xaa\xaa\x03\x00\x00\x00\x08\x06\x00\x01\x08\x00\x06\x04\x00\x01\x00\x19\xe3\xd3' \
'\x53\x52\xa9\xfe\xf7\x00\x00\x00\x00\x00\x00\x00\x43\x08\x0e\x36'
self.rt1 = RadioTap(self.frame_1)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_2 = str()
self.frame_2 += '\x00' # Header Revision
self.frame_2 += '\x00' # Header Pad
self.frame_2 += '\x24\x00' # Header Length
# Present Flags (tsft, flags, rate, channel, dbm antenna signal, rx flags, ns next, ext)
self.frame_2 += '\x2f\x40\x00\xa0'
self.frame_2 += '\x20\x08\x00\x00' # Present Flags (dbm antenna signal, antenna)
self.frame_2 += '\x00\x00\x00\x00' # ???
self.frame_2 += '\x97\xd3\x26\x44\x06\x00\x00\x00' # TSFT
self.frame_2 += '\x10' # Flags
self.frame_2 += '\x02' # Data Rate
self.frame_2 += '\x6c\x09' # Channel Frequency
self.frame_2 += '\xc0\x00' # Channel Type
self.frame_2 += '\xa6' # SSI Signal
self.frame_2 += '\x00' # ???
self.frame_2 += '\x00\x00' # RX flags
self.frame_2 += '\xa6' # SSI Signal
self.frame_2 += '\x00' # ???
self.frame_2 += '\xd4\x00\x00\x00\x9c\x04\xeb\x4d\xdb\x53\x8d\xf3\xc6\xc3'
self.rt2 = RadioTap(self.frame_2)
# RadioTap + 802.11 Data Frame + LLC SNAP + ARP Request
self.frame_3 = str()
self.frame_3 += '\x00' # Header Revision
self.frame_3 += '\x00' # Header Pad
self.frame_3 += '\x24\x00' # Header Length
# Present Flags (tsft, flags, rate, channel, dbm antenna signal, rx flags, ns next, ext)
self.frame_3 += '\x2f\x40\x00\xa0'
self.frame_3 += '\x20\x08\x00\x80' # Present Flags (dbm antenna signal, antenna)
self.frame_3 += '\x00\x00\x00\x00' # Present Flags
# self.frame_3 += '\x00\x00\x00\x00' # ???
self.frame_3 += '\x97\xd3\x26\x44\x06\x00\x00\x00' # TSFT
self.frame_3 += '\x10' # Flags
self.frame_3 += '\x02' # Data Rate
self.frame_3 += '\x6c\x09' # Channel Frequency
self.frame_3 += '\xc0\x00' # Channel Type
self.frame_3 += '\xa6' # SSI Signal
self.frame_3 += '\x00' # ???
self.frame_3 += '\x00\x00' # RX flags
self.frame_3 += '\xa6' # SSI Signal
self.frame_3 += '\x00' # ???
self.frame_3 += '\xd4\x00\x00\x00\x9c\x04\xeb\x4d\xdb\x53\x8d\xf3\xc6\xc3'
self.rt3 = RadioTap(self.frame_3)
def test_01_sizes(self):
"""Test RadioTap frame sizes"""
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_body_size(), len(self.frame_0) - 24)
self.assertEqual(self.rt0.get_tail_size(), 0)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_body_size(), len(self.frame_1) - 32)
self.assertEqual(self.rt1.get_tail_size(), 0)
def test_02_version(self):
"""Test RadioTap version getter/setter"""
self.assertEqual(self.rt0.get_version(), 0x00)
self.rt0.set_version(1)
self.assertEqual(self.rt0.get_version(), 0x01)
self.assertEqual(self.rt1.get_version(), 0x00)
self.rt1.set_version(1)
self.assertEqual(self.rt1.get_version(), 0x01)
def test_03_present(self):
"""Test RadioTap present getter"""
self.assertEqual(self.rt0.get_present(), 0x0000580e)
self.assertEqual(self.rt1.get_present(), 0x00040867)
def test_04_present_bits(self):
"""Test RadioTap present bits tester"""
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_TSFT), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_FLAGS), True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_RATE), True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_CHANNEL), True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_FHSS), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DBM_ANTSIGNAL),
False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DBM_ANTNOISE),
False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_LOCK_QUALITY),
False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_TX_ATTENUATION),
False)
self.assertEqual(
self.rt0.get_present_bit(RadioTap.RTF_DB_TX_ATTENUATION), False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DBM_TX_POWER),
False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_ANTENNA), True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DB_ANTSIGNAL),
True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DB_ANTNOISE),
False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_FCS_IN_HEADER),
True)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_TX_FLAGS),
False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_RTS_RETRIES),
False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_DATA_RETRIES),
False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_XCHANNEL),
False)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_EXT), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TSFT), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_FLAGS), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_RATE), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_CHANNEL), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_FHSS), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DBM_ANTSIGNAL),
True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DBM_ANTNOISE),
True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_LOCK_QUALITY),
False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TX_ATTENUATION),
False)
self.assertEqual(
self.rt1.get_present_bit(RadioTap.RTF_DB_TX_ATTENUATION), False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DBM_TX_POWER),
False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_ANTENNA), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DB_ANTSIGNAL),
False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DB_ANTNOISE),
False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_FCS_IN_HEADER),
False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TX_FLAGS),
False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_RTS_RETRIES),
False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_DATA_RETRIES),
False)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_XCHANNEL), True)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_EXT), False)
def test_05_tsft(self):
"""Test RadioTap tstf getter"""
self.assertEqual(self.rt0.get_tsft(), None)
self.assertEqual(self.rt1.get_tsft(), 622461744)
def test_06_tsft(self):
"""Test RadioTap tstf getter/setter"""
# When the field is new
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.rt0.set_tsft(0x0102030405060708)
self.assertEqual(self.rt0.get_tsft(), 0x0102030405060708)
self.assertEqual(self.rt0.get_header_size(), 24 + 8)
# When exist the field
self.rt0.set_tsft(0x0807060504030201)
self.assertEqual(self.rt0.get_tsft(), 0x0807060504030201)
self.assertEqual(self.rt0.get_header_size(), 24 + 8)
def test_07_unset_fields(self):
"""Test RadioTap unset field"""
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_FLAGS), True)
self.rt0.unset_field(RadioTap.RTF_FLAGS)
self.assertEqual(self.rt0.get_size(), len(self.frame_0) - 1)
self.assertEqual(self.rt0.get_header_size(), 24 - 1)
self.assertEqual(self.rt0.get_present_bit(RadioTap.RTF_FLAGS), False)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TSFT), True)
self.rt1.unset_field(RadioTap.RTF_TSFT)
self.assertEqual(self.rt1.get_size(), len(self.frame_1) - 8)
self.assertEqual(self.rt1.get_header_size(), 32 - 8)
self.assertEqual(self.rt1.get_present_bit(RadioTap.RTF_TSFT), False)
def test_08_flags_field(self):
"""Test RadioTap flags getter/setter"""
# When exist the field
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_flags(), 0x10)
self.rt0.set_flags(0xAB)
self.assertEqual(self.rt0.get_flags(), 0xAB)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_flags(), 0x22)
self.rt1.set_flags(0xAB)
self.assertEqual(self.rt1.get_flags(), 0xAB)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
# TODO: Test the size when the field is new
def test_09_rate_field(self):
"""Test RadioTap rate getter/setter"""
# When exist the field
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_rate(), 0x6c)
self.rt0.set_rate(0xAB)
self.assertEqual(self.rt0.get_rate(), 0xAB)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_rate(), 0x0c)
self.rt1.set_rate(0xAB)
self.assertEqual(self.rt1.get_rate(), 0xAB)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
# TODO: Test the size when the field is new
def test_10_channel_field(self):
"""Test RadioTap channel getter/setter"""
# When exist the field
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_channel(), (2412, 0x0480))
self.rt0.set_channel(freq=1234, flags=0x5678)
self.assertEqual(self.rt0.get_channel(), (1234, 0x5678))
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_channel(), None)
self.rt1.set_channel(freq=1234, flags=0x5678)
self.assertEqual(self.rt1.get_channel(), (1234, 0x5678))
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 4)
self.assertEqual(self.rt1.get_header_size(), 32 + 4)
def test_11_fhss_field(self):
"""Test RadioTap FHSS getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_FHSS(), None)
self.rt1.set_FHSS(hop_set=0xAB, hop_pattern=0xCD)
self.assertEqual(self.rt1.get_FHSS(), (0xAB, 0xCD))
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 2)
self.assertEqual(self.rt1.get_header_size(), 32 + 2)
def test_12_dbm_ant_signal_field(self):
"""Test RadioTap dBm Antenna Signal getter/setter"""
# When exist the field
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dBm_ant_signal(), 0xd9)
self.rt1.set_dBm_ant_signal(signal=0xF1)
self.assertEqual(self.rt1.get_dBm_ant_signal(), 0xF1)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
# When the field is new
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_dBm_ant_signal(), None)
self.rt0.set_dBm_ant_signal(signal=0xF1)
self.assertEqual(self.rt0.get_dBm_ant_signal(), 0xF1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0) + 1)
self.assertEqual(self.rt0.get_header_size(), 24 + 1)
def test_13_dbm_ant_noise_field(self):
"""Test RadioTap dBm Antenna Noise getter/setter"""
# When exist the field
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dBm_ant_noise(), 0xa0)
self.rt1.set_dBm_ant_noise(signal=0xF1)
self.assertEqual(self.rt1.get_dBm_ant_noise(), 0xF1)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
# When the field is new
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_dBm_ant_noise(), None)
self.rt0.set_dBm_ant_noise(signal=0xF1)
self.assertEqual(self.rt0.get_dBm_ant_noise(), 0xF1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0) + 1)
self.assertEqual(self.rt0.get_header_size(), 24 + 1)
def test_14_lock_quality_field(self):
"""Test RadioTap Lock Quality getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_lock_quality(), None)
self.rt1.set_lock_quality(quality=0xABBA)
self.assertEqual(self.rt1.get_lock_quality(), 0xABBA)
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 2)
self.assertEqual(self.rt1.get_header_size(), 32 + 2)
def test_15_tx_attenuation_field(self):
"""Test RadioTap Tx Attenuation getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_tx_attenuation(), None)
self.rt1.set_tx_attenuation(power=0xABBA)
self.assertEqual(self.rt1.get_tx_attenuation(), 0xABBA)
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 2)
self.assertEqual(self.rt1.get_header_size(), 32 + 2)
def test_16_dB_tx_attenuation_field(self):
"""Test RadioTap dB Tx Attenuation getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dB_tx_attenuation(), None)
self.rt1.set_dB_tx_attenuation(power=0xABBA)
self.assertEqual(self.rt1.get_dB_tx_attenuation(), 0xABBA)
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 2)
self.assertEqual(self.rt1.get_header_size(), 32 + 2)
def test_17_dbm_tx_power_field(self):
"""Test RadioTap dBm Tx Power getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dBm_tx_power(), None)
self.rt1.set_dBm_tx_power(power=-8)
self.assertEqual(self.rt1.get_dBm_tx_power(), -8)
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 1)
self.assertEqual(self.rt1.get_header_size(), 32 + 1)
def test_18_antenna_field(self):
"""Test RadioTap Antenna getter/setter"""
# TODO: When the field is new
# When exist the field
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_antenna(), 0x02)
self.rt1.set_antenna(antenna_index=0xF1)
self.assertEqual(self.rt1.get_antenna(), 0xF1)
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_antenna(), 0x00)
self.rt0.set_antenna(antenna_index=0xF1)
self.assertEqual(self.rt0.get_antenna(), 0xF1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
def test_19_db_ant_signal_field(self):
"""Test RadioTap dB Antenna Signal getter/setter"""
# When exist the field
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_dB_ant_signal(), 0x1e)
self.rt0.set_dB_ant_signal(signal=0xF1)
self.assertEqual(self.rt0.get_dB_ant_signal(), 0xF1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dB_ant_signal(), None)
self.rt1.set_dB_ant_signal(signal=0xF1)
self.assertEqual(self.rt1.get_dB_ant_signal(), 0xF1)
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 1)
self.assertEqual(self.rt1.get_header_size(), 32 + 1)
def test_20_db_ant_noise_field(self):
"""Test RadioTap dB Antenna Noise getter/setter"""
# TODO: When exist the field
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_dB_ant_noise(), None)
self.rt1.set_dB_ant_noise(signal=0xF1)
self.assertEqual(self.rt1.get_dB_ant_noise(), 0xF1)
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 1)
self.assertEqual(self.rt1.get_header_size(), 32 + 1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_dB_ant_noise(), None)
self.rt0.set_dB_ant_noise(signal=0xF1)
self.assertEqual(self.rt0.get_dB_ant_noise(), 0xF1)
self.assertEqual(self.rt0.get_size(), len(self.frame_0) + 1)
self.assertEqual(self.rt0.get_header_size(), 24 + 1)
# def test_21_rx_flags_field(self):
# """Test RadioTap RX Flags getter/setter"""
# # When the field is new
# # TODO: When exist the field
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
# self.assertEqual(self.rt2.get_header_size(),32)
# self.assertEqual(self.rt2.get_rx_flags(),None)
# self.rt2.set_rx_flags( signal=0xABBA )
# self.assertEqual(self.rt2.get_rx_flags(),0xABBA)
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+2)
# self.assertEqual(self.rt2.get_header_size(),32+2)
def test_22_fcs_in_header_field(self):
"""Test RadioTap FCS in header getter/setter"""
# When exist the field
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
self.assertEqual(self.rt0.get_FCS_in_header(), 0x00000000)
self.rt0.set_FCS_in_header(fcs=0x89ABCDEF)
self.assertEqual(self.rt0.get_FCS_in_header(), 0x89ABCDEF)
self.assertEqual(self.rt0.get_size(), len(self.frame_0))
self.assertEqual(self.rt0.get_header_size(), 24)
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_FCS_in_header(), None)
self.rt1.set_FCS_in_header(fcs=0x89ABCDEF)
self.assertEqual(self.rt1.get_FCS_in_header(), 0x89ABCDEF)
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 4)
self.assertEqual(self.rt1.get_header_size(), 32 + 4)
# def test_23_rssi_field(self):
# """Test RadioTap RSSI getter/setter"""
# # When the field is new
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
# self.assertEqual(self.rt2.get_header_size(),32)
# self.assertEqual(self.rt2.get_RSSI(),None)
# self.rt2.set_RSSI( rssi=0xBA, max_rssi=0xAB )
# self.assertEqual(self.rt2.get_RSSI(),( 0xBA, 0xAB))
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+2)
# self.assertEqual(self.rt2.get_header_size(),32+2)
# # TODO: When exist the field
def test_24_rts_retries_field(self):
"""Test RadioTap RTS retries getter/setter"""
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_RTS_retries(), None)
self.rt1.set_RTS_retries(retries=0xBA)
self.assertEqual(self.rt1.get_RTS_retries(), 0xBA)
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 1)
self.assertEqual(self.rt1.get_header_size(), 32 + 1)
# TODO: When exist the field
def test_25_tx_flags_field(self):
"""Test RadioTap TX flags getter/setter"""
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_tx_flags(), None)
self.rt1.set_tx_flags(flags=0xABBA)
self.assertEqual(self.rt1.get_tx_flags(), 0xABBA)
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 2)
self.assertEqual(self.rt1.get_header_size(), 32 + 2)
# TODO: When exist the field
def test_26_xchannel_field(self):
"""Test RadioTap xchannel getter/setter"""
(ch_type, ch_freq, ch_num, ch_maxpower) = self.rt1.get_xchannel()
self.assertEqual(ch_type, 0x00000140)
self.assertEqual(ch_freq, 5180)
self.assertEqual(ch_num, 36)
self.assertEqual(ch_maxpower, 0x11)
(ch_type, ch_freq, ch_num, ch_maxpower) = (0x12345678, 1234, 12, 34)
self.rt1.set_xchannel(flags=ch_type,
freq=ch_freq,
channel=ch_num,
maxpower=ch_maxpower)
(nch_type, nch_freq, nch_num, nch_maxpower) = self.rt1.get_xchannel()
self.assertEqual(ch_type, nch_type)
self.assertEqual(ch_freq, nch_freq)
self.assertEqual(ch_num, nch_num)
self.assertEqual(ch_maxpower, nch_maxpower)
def test_27_data_retries_field(self):
"""Test RadioTap Data retries getter/setter"""
# When the field is new
self.assertEqual(self.rt1.get_size(), len(self.frame_1))
self.assertEqual(self.rt1.get_header_size(), 32)
self.assertEqual(self.rt1.get_data_retries(), None)
self.rt1.set_data_retries(retries=0xAB)
self.assertEqual(self.rt1.get_data_retries(), 0xAB)
self.assertEqual(self.rt1.get_size(), len(self.frame_1) + 1)
self.assertEqual(self.rt1.get_header_size(), 32 + 1)
# TODO: When exist the field
# def test_28_hardware_queue_field(self):
# """Test RadioTap Hardware Queue getter/setter"""
# # When the field is new
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2))
# self.assertEqual(self.rt2.get_header_size(),32)
# self.assertEqual(self.rt2.get_hardware_queue(),None)
# self.rt2.set_hardware_queue( queue=0xAB )
# self.assertEqual(self.rt2.get_hardware_queue(),0xAB)
# self.assertEqual(self.rt2.get_size(),len(self.frame_orig_2)+1)
# self.assertEqual(self.rt2.get_header_size(),32+1)
# # TODO: When exist the field
def test_29_radiotap_length_field(self):
"""Test RadioTap header length field"""
# RadioTap from scratch calling get_length() and then get_packet()
rt = RadioTap()
# 0x08 bytes is the minimal headers size:
# 1 byte Revision
# 1 byte pad
# 2 bytes header length
# 4 bytes present flags
self.assertEqual(rt.get_header_length(), 0x08)
raw_packet = rt.get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00")
# RadioTap from scratch without call to get_length()
raw_packet = RadioTap().get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00")
def test_30_radiotap_length_filed_with_payload(self):
"""Test RadioTap header length field with payload"""
# RadioTap from scratch calling get_length() and then get_packet()
rt = RadioTap()
self.assertEqual(rt.get_header_length(), 0x08)
data = Data("aa")
rt.contains(data)
self.assertEqual(rt.get_header_length(),
0x08) # The header length is the same
raw_packet = rt.get_packet()
self.assertEqual(raw_packet, "\x00\x00\x08\x00\x00\x00\x00\x00aa")
def test_31_radiotap_present_flags_extended(self):
"""Test RadioTap extended present flags"""
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_EXT), True)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_RATE), True)
self.assertEqual(self.rt2.get_present_bit(RadioTap.RTF_CHANNEL), True)
self.assertEqual(self.rt2.get_channel(), (2412, 192))
self.assertEqual(self.rt2.get_rate(), 2)
self.assertEqual(self.rt2.get_dBm_ant_signal(), 0xa6)
self.assertEqual(self.rt3.get_present_bit(RadioTap.RTF_EXT), True)
self.assertEqual(self.rt3.get_present_bit(RadioTap.RTF_RATE), True)
self.assertEqual(self.rt3.get_present_bit(RadioTap.RTF_CHANNEL), True)
self.assertEqual(self.rt3.get_channel(), (2412, 192))
self.assertEqual(self.rt3.get_rate(), 2)
self.assertEqual(self.rt3.get_dBm_ant_signal(), 0xa6)