def testCorruptFiles(self):
signature = 'wifi4|probe:0|assoc:1,htcap:this_is_not_a_number'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertIn('802.11n', taxonomy[2])
signature = 'wifi4|probe:0|assoc:1,vhtcap:this_is_not_a_number'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertIn('802.11ac', taxonomy[2])
def testRealClientsPerformance(self):
"""Test the performance information for a few real clients."""
# Nest Thermostat
sig = (
'wifi4|probe:0,1,50,45,htcap:0130,htagg:18,htmcs:000000ff|assoc:'
'0,1,50,48,45,221(0050f2,2),htcap:013c,htagg:18,htmcs:000000ff')
taxonomy = wifi.identify_wifi_device(sig, '18:b4:30:00:00:01')
self.assertEqual('802.11n n:1,w:20', taxonomy[2])
# Samsung Galaxy S4
sig = (
'wifi4|probe:0,1,45,127,191,221(001018,2),221(00904c,51),221(00904c,'
'4),221(0050f2,8),htcap:006f,htagg:17,htmcs:000000ff,vhtcap:0f805832,'
'vhtrxmcs:0000fffe,vhttxmcs:0000fffe|assoc:0,1,33,36,48,45,127,191,'
'221(001018,2),221(00904c,4),221(0050f2,2),htcap:006f,htagg:17,htmcs:'
'000000ff,vhtcap:0f805832,vhtrxmcs:0000fffe,vhttxmcs:0000fffe')
taxonomy = wifi.identify_wifi_device(sig, 'cc:3a:61:00:00:01')
self.assertEqual('802.11ac n:1,w:80', taxonomy[2])
# MacBook Pro 802.11ac
sig = (
'wifi4|probe:0,1,45,127,191,221(00904c,51),htcap:09ef,htagg:17,'
'htmcs:0000ffff,vhtcap:0f8259b2,vhtrxmcs:0000ffea,vhttxmcs:0000ffea|'
'assoc:0,1,33,36,48,45,127,191,221(00904c,51),221(0050f2,2),htcap:09ef,'
'htagg:17,htmcs:0000ffff,vhtcap:0f8259b2,vhtrxmcs:0000ffea,'
'vhttxmcs:0000ffea')
taxonomy = wifi.identify_wifi_device(sig, '3c:15:c2:00:00:01')
self.assertEqual('802.11ac n:3,w:80', taxonomy[2])
def testNameLookup(self):
signature = (
'wifi4|probe:0,1,45,3,221(001018,2),221(00904c,51),htcap:0100,'
'htagg:19,htmcs:000000ff|assoc:0,1,33,36,48,45,221(001018,2),'
'221(00904c,51),221(0050f2,2),htcap:0100,htagg:19,'
'htmcs:000000ff,txpow:180f')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual(3, len(taxonomy))
self.assertEqual('Unknown', taxonomy[1])
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual(3, len(taxonomy))
self.assertEqual('Unknown', taxonomy[1])
taxonomy = wifi.identify_wifi_device(signature, 'c8:69:cd:5e:b5:43')
self.assertEqual('Apple TV', taxonomy[0])
self.assertEqual('3rd gen', taxonomy[1])
def testPerformanceInfoBroken(self):
signature = ('wifi4|probe:0,htmcs:000000ff|assoc:0,htmcs:000000ff')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11a/b/g n:1,w:20', taxonomy[2])
signature = ('wifi4|probe:0,htcap:wrong,htmcs:ffffffff|'
'assoc:0,htcap:wrong,htmcs:ffffffff')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11n n:4,w:??', taxonomy[2])
signature = ('wifi4|probe:0,htcap:012c,htmcs:wrong|'
'assoc:0,htcap:012c,htmcs:wrong')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11n n:?,w:20', taxonomy[2])
signature = ('wifi4|probe:0,htcap:wrong,htmcs:wrong|'
'assoc:0,htcap:wrong,htmcs:wrong')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11n n:?,w:??', taxonomy[2])
def test802_11n_NssWidth(self):
signature = 'wifi4|probe:0|assoc:1,htcap:012c,htagg:03,htmcs:000000ff'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11n n:1,w:20', taxonomy[2])
signature = 'wifi4|probe:0|assoc:1,htcap:0102,htagg:03,htmcs:0000ffff'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11n n:2,w:40', taxonomy[2])
signature = 'wifi4|probe:0|assoc:1,htcap:0200,htagg:03,htmcs:00ffffff'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11n n:3,w:20', taxonomy[2])
signature = 'wifi4|probe:0|assoc:1,htcap:0302,htagg:03,htmcs:ffffffff'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11n n:4,w:40', taxonomy[2])
signature = 'wifi4|probe:0|assoc:1'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11a/b/g n:1,w:20', taxonomy[2])
def testUnknown(self):
signature = 'wifi4|probe:0,1,2,vhtcap:0033|assoc:3,4,vhtcap:0033'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertIn('802.11ac', taxonomy[2])
self.assertNotIn('802.11n', taxonomy[2])
self.assertNotIn('802.11a/b/g', taxonomy[2])
signature = 'wifi4|probe:0,1,2,htcap:0033|assoc:3,4,htcap:0033'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertNotIn('802.11ac', taxonomy[2])
self.assertIn('802.11n', taxonomy[2])
self.assertNotIn('802.11a/b/g', taxonomy[2])
signature = 'wifi4|probe:0,1,2|assoc:3,4'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertNotIn('802.11ac', taxonomy[2])
self.assertNotIn('802.11n', taxonomy[2])
self.assertIn('802.11a/b/g', taxonomy[2])
def testCommonSignature(self):
signature = ('wifi4|probe:0,1,50,45,221(001018,2),221(00904c,51),'
'htcap:182c,htagg:1b,htmcs:000000ff|assoc:0,1,48,50,45,'
'221(001018,2),221(00904c,51),221(0050f2,2),'
'htcap:182c,htagg:1b,htmcs:000000ff')
# This is a very common signature among devices issued in 2010.
# It will match:
# Samsung Captivate, Epic 2, Fascinate, Continuum, Charge, Vibrant
# Samsung Galaxy Tab 2
# HTC Thunderbolt
# ASUS Transformer TF300
# Nexus One
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertIn('Unknown', taxonomy[1])
taxonomy = wifi.identify_wifi_device(signature, 'b4:ce:f6:00:00:01')
self.assertIn('Unknown', taxonomy[1])
taxonomy = wifi.identify_wifi_device(signature, 'ac:22:0b:00:00:01')
self.assertIn('Unknown', taxonomy[1])
def check_pcap(expected_model, pcap, ap_mode=False):
mac, sig = get_taxonomy_from_pcap(pcap, ap_mode)
genus, species, _ = wifi.identify_wifi_device(sig, mac)
actual_model = genus + " " + species if species else genus
if expected_model and expected_model != actual_model:
print 'Mismatch in %s: %s %s != %s' % (pcap, mac, expected_model,
actual_model)
return True
if not expected_model and 'Unknown' not in actual_model:
print 'Mismatch in %s: %s %s != Unknown' % (pcap, mac, actual_model)
return True
def testLookup(self):
signature = ('wifi4|probe:0,1,50,45,htcap:186e|assoc:0,1,50,48,'
'221(0050f2,2),45,127,htcap:086c,htmcs:000000ff')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual(3, len(taxonomy))
self.assertEqual('802.11n n:1,w:20', taxonomy[2])
signature = ('wifi4|probe:0,1,45,191,221(0050f2,4),221(506f9a,9),'
'221(001018,2),htcap:006f,htagg:17,htmcs:0000ffff,'
'vhtcap:0f815832,vhtrxmcs:0000fffa,vhttxmcs:0000fffa,'
'extcap:000008800140,wps:Nexus_6|assoc:0,1,33,36,48,45,'
'127,191,221(001018,2),221(0050f2,2),htcap:006f,htagg:17,'
'htmcs:0000ffff,vhtcap:0f815832,vhtrxmcs:0000fffa,'
'vhttxmcs:0000fffa,txpow:e009,extcap:000008800140')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:02')
expected = 'Nexus 6;;802.11ac n:2,w:80'
self.assertEqual(';'.join(taxonomy), expected)
self.assertEqual(3, len(taxonomy))
self.assertEqual('Nexus 6', taxonomy[0])
self.assertEqual('', taxonomy[1])
self.assertEqual('802.11ac n:2,w:80', taxonomy[2])
def testOUI(self):
# Devices with a generic signature, distinguished via MAC OUI
signature = (
'wifi4|probe:0,1,50,3,45,221(0050f2,8),htcap:012c,htagg:03,'
'htmcs:000000ff|assoc:0,1,50,33,48,70,45,221(0050f2,2),127,'
'htcap:012c,htagg:03,htmcs:000000ff,txpow:170d,'
'extcap:00000a0200000000')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertIn('Unknown', taxonomy[1])
taxonomy = wifi.identify_wifi_device(signature, 'b4:52:7e:00:00:01')
self.assertIn('Sony Xperia', taxonomy[0])
taxonomy = wifi.identify_wifi_device(signature, 'f8:f1:b6:00:00:01')
self.assertIn('Moto E', taxonomy[0])
self.assertIn('2nd gen', taxonomy[1])
# Test one of the OUIs with multiple vendors listed.
signature = (
'wifi4|probe:0,1,3,45,50,htcap:0120,htagg:03,htmcs:00000000|'
'assoc:0,1,48,50,127,221(0050f2,2),45,htcap:012c,'
'htagg:03,htmcs:000000ff,extcap:0000000000000140|oui:google')
taxonomy = wifi.identify_wifi_device(signature, '6c:ad:f8:00:00:01')
self.assertEqual('Chromecast', taxonomy[0])
self.assertEqual('v1', taxonomy[1])
def testBrokenNssWidth(self):
"""Test for broken client behavior.
A few clients, notably Nexus 4 with Android 4.2,
include a VHT Capabilities in their Probe even
though they are not 802.11ac devices. Presumably
the driver supports other chipsets which are.
To work around this, taxonomy is only supposed to
look at the Association for determining client
performance characteristics.
"""
signature = ('wifi4|probe:0,1,50,45,221(0050f2,8),191,221(0050f2,4),'
'221(506f9a,9),htcap:012c,htagg:03,htmcs:000000ff,'
'vhtcap:31811120,vhtrxmcs:01b2fffc,vhttxmcs:01b2fffc,'
'wps:Nexus_4|assoc:0,1,50,48,45,221(0050f2,2),'
'htcap:012c,htagg:03,htmcs:000000ff')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11n n:1,w:20', taxonomy[2])
def test802_11ac_Width(self):
signature = ('wifi4|probe:0|assoc:1,htcap:0302,htmcs:000000ff,'
'vhtcap:00000000,vhtrxmcs:0000ffaa,vhttxmcs:0000ffaa')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11ac n:4,w:80', taxonomy[2])
signature = ('wifi4|probe:0|assoc:1,htcap:0200,htmcs:000000ff,'
'vhtcap:00000004,vhtrxmcs:0000ffea,vhttxmcs:0000ffea')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11ac n:3,w:160', taxonomy[2])
signature = ('wifi4|probe:0|assoc:1,htcap:0200,htmcs:000000ff,'
'vhtcap:00000004,vhtrxmcs:0000fffa,vhttxmcs:0000fffa')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11ac n:2,w:160', taxonomy[2])
signature = ('wifi4|probe:0|assoc:1,htcap:0200,htmcs:000000ff,'
'vhtcap:00000004,vhtrxmcs:0000fffe,vhttxmcs:0000fffe')
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11ac n:1,w:160', taxonomy[2])
signature = 'wifi4|probe:0|assoc:1,vhtcap:00000008'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11ac n:?,w:80+80', taxonomy[2])
signature = 'wifi4|probe:0|assoc:1,vhtcap:0000000c'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertEqual('802.11ac n:?,w:??', taxonomy[2])
def testOS(self):
signature = 'wifi4|probe:0,1,50|assoc:0,1,50,48,221(0050f2,2)'
taxonomy = wifi.identify_wifi_device(signature, '00:00:01:00:00:01')
self.assertIn('Unknown', taxonomy[1])
taxonomy = wifi.identify_wifi_device(signature, '28:ef:01:00:00:01')
self.assertIn('Kindle', taxonomy[0])
def testChecksumWhenNoIdentification(self):
taxonomy = wifi.identify_wifi_device(
'wifi|probe:1,2,3,4,htcap:0|assoc:1', '00:00:01:00:00:01')
h = 'SHA:27b78dbb1bc795961ddad0686137eb9fddbbc7f8766bd8947b4deca563b830be'
self.assertIn(h, taxonomy[0])