Merge pull request #105 from tiagogalvao/tests

Tests

Author: kimocoder

requirements.txt

@@ -1 +1,4 @@
+chardet
 pytest-flake8
+scapy
+argparse~=1.2.1

runtests.sh

@@ -4,8 +4,10 @@
 #sleep 2
 #sh tests/test_flake8.sh
 #sleep 2
+clear
+
 echo "** Starting unittest/pytest .."
-sleep 5
+sleep 2
 python3 -m unittest discover tests -v
 sleep 2
 echo "** runtests.sh test script at the end"

tests/files/airodump.csv

@@ -22,14 +22,14 @@ F2:64:20:05:BF:90, 2015-05-30 11:28:43, 2015-05-30 11:28:50,  6,  54, WPA2, CCMP
 00:1D:D5:9B:11:00, 2015-05-30 11:28:43, 2015-05-30 11:28:50,  6,  54, WPA2, CCMP TKIP,PSK, -52,       32,        6,   0.  0.  0.  0,   9, HOME-1102, 
 00:24:7B:AB:5C:EE, 2015-05-30 11:28:43, 2015-05-30 11:28:50,  6,  54, WPA2 WPA, CCMP TKIP,PSK, -55,       16,       25,   0.  0.  0.  0,  11, myqwest0445, 
 02:1D:D5:9B:11:00, 2015-05-30 11:28:43, 2015-05-30 11:28:50,  6,  54, WPA2, CCMP,PSK, -53,       41,        0,   0.  0.  0.  0,   0, , 
-06:1D:D5:9B:11:00, 2015-05-30 11:28:43, 2015-05-30 11:28:50,  6,  54, OPN, ,   , -53,       56,        0,   0.  0.  0.  0,  11, xfinitywifi, 
+06:1D:D5:9B:11:00, 2015-05-30 11:28:43, 2015-05-30 11:28:50,  6,  54, OPN, ,   , -53,       56,        0,   0.  0.  0.  0,  9, 这种情威灵顿酒店À,
 00:F7:6F:CD:B2:2A, 2015-05-30 11:28:43, 2015-05-30 11:28:50,  6,  54, WPA2, CCMP,PSK, -48,       44,        0,   0.  0.  0.  0,  13, im human 2015, 
 30:85:A9:39:D2:18, 2015-05-30 11:28:43, 2015-05-30 11:28:50,  6,  54, WPA2, CCMP,PSK, -21,       44,        4,   0.  0.  0.  0,  32, Uncle Router's Gigabit LAN Party, 
 00:0E:58:FA:7C:61, 2015-05-30 11:28:44, 2015-05-30 11:28:49,  6,  -1, WPA, ,   ,  -1,        0,       57,   0.  0.  0.  0,   0, , 
 00:0E:58:F8:0B:B5, 2015-05-30 11:28:44, 2015-05-30 11:28:48,  6,  -1, WPA, ,   ,  -1,        0,       59,   0.  0.  0.  0,   0, , 
 28:01:00:00:D0:00, 2015-05-30 11:28:44, 2015-05-30 11:28:44,  6,  -1, , ,   ,  -1,        0,        0,   0.  0.  0.  0,   0, , 
 00:0E:58:E9:36:B3, 2015-05-30 11:28:44, 2015-05-30 11:28:48,  6,  -1, WPA, ,   ,  -1,        0,        2,   0.  0.  0.  0,   0, , 
-05:00:40:00:BB:7C, 2015-05-30 11:28:50, 2015-05-30 11:28:50, -1,  -1, , ,   ,  -1,        0,        0,   0.  0.  0.  0,   0, , 
+05:00:40:00:BB:7C, 2015-05-30 11:28:50, 2015-05-30 11:28:50, -1,  -1, , ,   ,  -1,        0,        0,   0.  0.  0.  0,   0, ,
 
 Station MAC, First time seen, Last time seen, Power, # packets, BSSID, Probed ESSIDs
 3A:01:44:32:C8:5C, 2015-05-30 11:28:44, 2015-05-30 11:28:44, -69,        3, 28:01:00:00:D0:00,

tests/test_Airmon.py

@@ -1,13 +1,10 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
-
 import sys
-
-sys.path.insert(0, '..')
-
+import unittest
 from wifite.tools.airmon import Airmon
 
-import unittest
+sys.path.insert(0, '..')
 
 
 class TestAirmon(unittest.TestCase):

tests/test_Airodump.py

@@ -2,17 +2,26 @@
 # -*- coding: utf-8 -*-
 
 import sys
-
-sys.path.insert(0, '..')
-
 from wifite.tools.airodump import Airodump
-
 import unittest
 
+sys.path.insert(0, '..')
+
 
 class TestAirodump(unittest.TestCase):
     """ Test suite for Wifite's interaction with the Airodump tool """
 
+    def test_airodump(self):
+        csv_filename = self.getFile('airodump.csv')
+        targets = Airodump.get_targets_from_csv(csv_filename)
+
+        print('')
+        for target in targets:
+            print("Testing ESSID: ", target.essid)
+            if target.essid is not None:
+                assert target.essid_len == len(target.essid), \
+                    f'ESSID length is {target.essid_len} but ESSID is {len(target.essid)} - [{target.essid}]'
+
     def test_airodump_weird_characters(self):
         csv_filename = self.getFile('airodump-weird-ssids.csv')
         targets = Airodump.get_targets_from_csv(csv_filename)
@@ -35,11 +44,10 @@ def test_airodump_weird_characters(self):
 
         # Hidden access point
         target = targets[4]
-        assert target.essid_known == False, 'ESSID full of null characters should not be known'
+        assert target.essid_known is False, 'ESSID full of null characters should not be known'
         expected = None
         assert target.essid == expected, f'Expected ESSID ({expected}) but got ({target.essid})'
-
-        assert target.essid_len == 19, f'ESSID length shold be 19, but got {target.essid_len}'
+        assert target.essid_len == 19, f'ESSID [unknow chars] length should be 19, but got {target.essid_len}'
 
     def getFile(self, filename):
         """ Helper method to parse targets from filename """

tests/test_Handshake.py

@@ -2,21 +2,21 @@
 # -*- coding: utf-8 -*-
 
 import sys
+import unittest
 
 from wifite.model.handshake import Handshake
 from wifite.util.process import Process
 
 sys.path.insert(0, '..')
 
-import unittest
-
 
 class TestHandshake(unittest.TestCase):
     """ Test suite for Target parsing an generation """
 
     def getFile(self, filename):
         """ Helper method to parse targets from filename """
-        import os, inspect
+        import os
+        import inspect
         this_file = os.path.abspath(inspect.getsourcefile(self.getFile))
         this_dir = os.path.dirname(this_file)
         return os.path.join(this_dir, 'files', filename)
@@ -31,28 +31,32 @@ def testAnalyze(self):
 
     @unittest.skipUnless(Process.exists("tshark"), 'tshark is missing')
     def testHandshakeTshark(self):
+        print("\nTesting handshake with tshark...")
         hs_file = self.getFile("handshake_exists.cap")
         hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
-        assert (len(hs.tshark_handshakes()) > 0)
+        assert (len(hs.tshark_handshakes()) > 0), f'Expected len>0 but got len({len(hs.tshark_handshakes())})'
 
+    # TODO: Pyrit is broken
     @unittest.skipUnless(Process.exists("pyrit"), 'pyrit is missing')
     def testHandshakePyrit(self):
+        print("\nTesting handshake with pyrit...")
         hs_file = self.getFile("handshake_exists.cap")
         hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
-        assert (len(hs.pyrit_handshakes()) > 0)
+        assert (len(hs.pyrit_handshakes()) > 0), f'Expected len>0 but got len({len(hs.pyrit_handshakes())})'
 
     @unittest.skipUnless(Process.exists("cowpatty"), 'cowpatty is missing')
     def testHandshakeCowpatty(self):
+        print("\nTesting handshake with cowpatty...")
         hs_file = self.getFile('handshake_exists.cap')
         hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
-        hs.divine_bssid_and_essid()
-        assert (len(hs.cowpatty_handshakes()) > 0)
+        assert (len(hs.cowpatty_handshakes()) > 0), f'Expected len>0 but got len({len(hs.cowpatty_handshakes())})'
 
     @unittest.skipUnless(Process.exists("aircrack-ng"), 'aircrack-ng is missing')
     def testHandshakeAircrack(self):
+        print("\nTesting handshake with aircrack-ng...")
         hs_file = self.getFile('handshake_exists.cap')
         hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
-        assert (len(hs.aircrack_handshakes()) > 0)
+        assert (len(hs.aircrack_handshakes()) > 0), f'Expected len>0 but got len({len(hs.aircrack_handshakes())})'
 
 
 if __name__ == '__main__':

tests/test_Target.py

@@ -2,18 +2,17 @@
 # -*- coding: utf-8 -*-
 
 import unittest
-
 from wifite.tools.airodump import Airodump
 
 
 class TestTarget(unittest.TestCase):
     """ Test suite for Target parsing an generation """
-
     airodump_csv = 'airodump.csv'
 
     def getTargets(self, filename):
         """ Helper method to parse targets from filename """
-        import os, inspect
+        import os
+        import inspect
         this_file = os.path.abspath(inspect.getsourcefile(TestTarget.getTargets))
         this_dir = os.path.dirname(this_file)
         csv_file = os.path.join(this_dir, 'files', filename)

tools/pyrit/cpyrit/pckttools.py

@@ -77,14 +77,18 @@ def isFlagSet(self, name, value):
     """
     field, val = self.getfield_and_val(name)
     if isEnumField(field):
-        if val not in field.i2s:
-            return False
-        return field.i2s[val] == value
+        if field.i2s is not None:
+            if val not in field.i2s:
+                return False
+            return field.i2s[val] == value
+
     else:
         try:
             return (1 << field.names.index(value)) & self.__getattr__(name) != 0
         except:
             return (1 << field.names.index([value])) & self.__getattr__(name) != 0
+
+
 scapy.packet.Packet.isFlagSet = isFlagSet
 del isFlagSet
 
@@ -599,12 +603,15 @@ def open_offline(self, fname):
 
     def read(self):
         """Read one packet from the capture-source."""
-        r = _cpyrit_cpu.PcapDevice.read(self)
-        if r is not None:
-            ts, pckt_string = r
-            pckt = self.datalink_handler(pckt_string)
-            return pckt
-        else:
+        try:
+            r = _cpyrit_cpu.PcapDevice.read(self)
+            if r is not None:
+                ts, pckt_string = r
+                pckt = self.datalink_handler(pckt_string)
+                return pckt
+            else:
+                return None
+        except:
             return None
 
     def __iter__(self):

tools/pyrit/pyrit

@@ -5,6 +5,6 @@ if __name__ == "__main__":
     try:
         pyrit_cli.Pyrit_CLI().initFromArgv()
     except (KeyboardInterrupt, SystemExit):
-        print( >> sys.stderr, "\nInterrupted..."
-    except pyrit_cli.PyritRuntimeError, e:
+        print("\nInterrupted...")
+    except pyrit_cli.PyritRuntimeError as e:
         sys.exit(e)