Initial commit, basic helper classes created

py/Interface.py

@@ -0,0 +1,70 @@
+#!/usr/bin/python
+
+from Color import Color
+
+class Interface(object):
+    '''
+        Represents an 'interface' known by airmon-ng
+    '''
+
+    # Max length of fields.
+    # Used for printing a table of interfaces.
+    PHY_LEN = 6
+    NAME_LEN = 12
+    DRIVER_LEN = 12
+    CHIPSET_LEN = 30
+
+    def __init__(self, fields):
+        '''
+            Initializes & stores info about an interface.
+
+            Args:
+                Fields - list of fields
+                        0: PHY
+                        1: NAME
+                        2: DRIVER
+                        3: CHIPSET
+        '''
+        if len(fields) == 3:
+            fields.insert(0, 'phyX')
+        if len(fields) != 4:
+            raise Exception("Expected 4, got %d in %s" % (len(fields), fields))
+        self.phy = fields[0].strip()
+        self.name = fields[1].strip()
+        self.driver = fields[2].strip()
+        self.chipset = fields[3].strip()
+
+    def __str__(self):
+        ''' Colored string representation of interface '''
+        s = Color.s("{W}%s" % self.phy)
+        s += ' ' * max(Interface.PHY_LEN - len(self.phy), 0)
+
+        s += Color.s("{G}%s" % self.name)
+        s += ' ' * max(Interface.NAME_LEN - len(self.name), 0)
+
+        s += Color.s("{C}%s" % self.driver)
+        s += ' ' * max(Interface.DRIVER_LEN - len(self.driver), 0)
+
+        s += Color.s("{W}%s" % self.chipset)
+        s += ' ' * max(Interface.CHIPSET_LEN - len(self.chipset), 0)
+        return s
+
+    @staticmethod
+    def menu_header():
+        ''' Colored header row for interfaces '''
+        s = '    '
+        s += 'PHY'
+        s += ' ' * (Interface.PHY_LEN - len("PHY"))
+
+        s += 'Interface'
+        s += ' ' * (Interface.NAME_LEN - len("Interface"))
+        s += 'Driver'
+        s += ' ' * (Interface.DRIVER_LEN - len("Driver"))
+
+        s += 'Chipset'
+        s += ' ' * (Interface.CHIPSET_LEN - len("Chipset"))
+
+        s += '\n---'
+        s += '-' * (Interface.PHY_LEN + Interface.NAME_LEN + Interface.DRIVER_LEN + Interface.CHIPSET_LEN)
+        return s
+