1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
#!/usr/bin/env python
# Impacket - Collection of Python classes for working with network protocols.
#
# Copyright Fortra, LLC and its affiliated companies
#
# All rights reserved.
#
# This software is provided under a slightly modified version
# of the Apache Software License. See the accompanying LICENSE file
# for more information.
#
import unittest
from six import PY2
from impacket.ImpactDecoder import Dot11Decoder #,Dot11Types
class TestDot11Decoder(unittest.TestCase):
def setUp(self):
self.WEPKey=None #Unknown
self.WEPData=b'\x08\x41\x3a\x01\x00\x17\x3f\x44\x4f\x96\x00\x13\xce\x67\x0e\x73\x00\x17\x3f\x44\x4f\x96\xb0\x04\xeb\xcd\x8b\x00\x6e\xdf\x93\x36\x39\x5a\x39\x66\x6b\x96\xd1\x7a\xe1\xae\xb6\x11\x22\xfd\xf0\xd4\x0d\x6a\xb8\xb1\xe6\x2e\x1f\x25\x7d\x64\x1a\x07\xd5\x86\xd2\x19\x34\xb5\xf7\x8a\x62\x33\x59\x6e\x89\x01\x73\x50\x12\xbb\xde\x17\xdd\xb5\xd4\x35'
dot11_decoder = Dot11Decoder()
self.in0=dot11_decoder.decode(self.WEPData)
self.in1=self.in0.child()
self.in2=self.in1.child()
self.in3=self.in2.child()
if self.WEPKey:
self.in4=self.in3.child()
self.in5=self.in4.child()
def test_01_Dot11Decoder(self):
'Test Dot11 decoder'
if PY2:
self.assertEqual(str(self.in0.__class__), "impacket.dot11.Dot11")
else:
self.assertEqual(str(self.in0.__class__), "<class 'impacket.dot11.Dot11'>")
def test_02_Dot11DataFrameDecoder(self):
'Test Dot11DataFrame decoder'
if PY2:
self.assertEqual(str(self.in1.__class__), "impacket.dot11.Dot11DataFrame")
else:
self.assertEqual(str(self.in1.__class__), "<class 'impacket.dot11.Dot11DataFrame'>")
def test_03_Dot11WEP(self):
'Test Dot11WEP decoder'
if PY2:
self.assertEqual(str(self.in2.__class__), "impacket.dot11.Dot11WEP")
else:
self.assertEqual(str(self.in2.__class__), "<class 'impacket.dot11.Dot11WEP'>")
def test_04_Dot11WEPData(self):
'Test Dot11WEPData decoder'
if not self.WEPKey:
return
self.assertEqual(str(self.in3.__class__), "impacket.dot11.Dot11WEPData")
# Test if wep data "get_packet" is correct
wepdata=b'\x6e\xdf\x93\x36\x39\x5a\x39\x66\x6b\x96\xd1\x7a\xe1\xae\xb6\x11\x22\xfd\xf0\xd4\x0d\x6a\xb8\xb1\xe6\x2e\x1f\x25\x7d\x64\x1a\x07\xd5\x86\xd2\x19\x34\xb5\xf7\x8a\x62\x33\x59\x6e\x89\x01\x73\x50\x12\xbb\xde\x17'
self.assertEqual(self.in3.get_packet(), wepdata)
def test_05_LLC(self):
'Test LLC decoder'
if self.WEPKey:
self.assertEqual(str(self.in4.__class__), "impacket.dot11.LLC")
def test_06_Data(self):
'Test LLC Data decoder'
if self.WEPKey:
dataclass=self.in4.__class__
else:
dataclass=self.in3.__class__
self.assertGreater(str(dataclass).find('ImpactPacket.Data'), 0)
if __name__ == '__main__':
unittest.main(verbosity=1)
|
