Add WPS algo suggestions

drygdryg · Jan 1, 2020 · ebd5e9a · ebd5e9a
1 parent 7f7dc98
commit ebd5e9a
Showing 1 changed file with 82 additions and 2 deletions.
diff --git a/WPSpin.py b/WPSpin.py
@@ -53,11 +53,16 @@ def __init__(self):
         self.algos['pinH108L'] = {'name': 'H108L', 'mode': self.ALGO_STATIC, 'gen': lambda mac: 9422988}
         self.algos['pinONO'] = {'name': 'CBN ONO', 'mode': self.ALGO_STATIC, 'gen': lambda mac: 9575521}
 
-    def _parseMac(self, mac):
+    def _parseMAC(self, mac):
         mac = mac.replace(':', '').replace('-', '').replace('.', '')
         mac = int(mac, 16)
         return mac
 
+    def _parseOUI(self, mac):
+        mac = mac.replace(':', '').replace('-', '').replace('.', '')
+        oui = int(mac[:6], 16)
+        return oui
+
     def checksum(self, pin):
         '''
         Standard WPS checksum algorithm.
@@ -78,7 +83,7 @@ def generate(self, algo, mac):
         @algo — the WPS pin algorithm ID
         Returns the WPS pin string value
         '''
-        mac = self._parseMac(mac)
+        mac = self._parseMAC(mac)
         if algo not in self.algos:
             raise WPSException('Invalid WPS pin algorithm')
         pin = self.algos[algo]['gen'](mac)
@@ -107,13 +112,88 @@ def getAll(self, mac, get_static=True):
         return res
 
     def getList(self, mac, get_static=True):
+        '''
+        Get all WPS pin's for single MAC as list
+        '''
         res = []
         for ID, algo in self.algos.items():
             if algo['mode'] == self.ALGO_STATIC and not get_static:
                 continue
             res.append(self.generate(ID, mac))
         return res
 
+    def getSuggested(self, mac):
+        '''
+        Get all suggested WPS pin's for single MAC
+        '''
+        algos = self.suggest(mac)
+        res = []
+        for ID in algos:
+            algo = self.algos[ID]
+            item = {}
+            item['id'] = ID
+            if algo['mode'] == self.ALGO_STATIC:
+                item['name'] = 'Static PIN — ' + algo['name']
+            else:
+                item['name'] = algo['name']
+            item['pin'] = self.generate(ID, mac)
+            res.append(item)
+        return res
+
+    def getSuggestedList(self, mac):
+        '''
+        Get all suggested WPS pin's for single MAC as list
+        '''
+        algos = self.suggest(mac)
+        res = []
+        for algo in algos:
+            res.append(self.generate(algo, mac))
+        return res
+
+    def suggest(self, mac):
+        '''
+        Get algos suggestions for single MAC
+        Returns the algo ID
+        '''
+        oui = self._parseOUI(mac)
+        algorithms = {
+            'pin24': (3318, 5329, 7391, 7967, 8821, 8951, 9819, 9934, 26187, 40998, 45068, 57376, 311149, 528501, 555596, 558651, 941390, 1080303, 1354211, 1358184, 1365581, 1867493, 2099437, 2108353, 2150192, 2631773, 2762845, 3180336, 3322588, 3435475, 3455642, 3676006, 4213251, 5021439, 5041630, 5135694, 5269488, 6048937, 6071197, 6091947, 6431549, 6438116, 6438399, 6443988, 6444444, 6450131, 6454622, 6461119, 6465764, 6469254, 6471791, 6473247, 6473492, 6474664, 6475198, 6482043, 6559472, 6582274, 6862543, 6954343, 6955837, 6957149, 6962687, 6968276, 6968732, 6974419, 6978910, 6985407, 6990052, 6996079, 6997535, 6997780, 6998952, 6999486, 7000414, 7000423, 7478631, 7480125, 7486692, 7486975, 7492564, 7493020, 7498707, 7503198, 7509695, 7514340, 7520367, 7521823, 7522068, 7523240, 7523774, 7524702, 7530619, 7891593, 7900663, 8396546, 8702386, 8971179, 9329998, 9475300, 9496250, 10000337, 10548161, 11151453, 11552890, 11580124, 11683580, 11834446, 12325889, 12383877, 12888093, 13122101, 13134983, 13393230, 13524302, 13921884, 13942655, 14183657, 14216087, 14446916, 14696758, 14732110, 14827830, 14828534, 14974201, 15256877, 15345757, 15475517, 15483894, 15518512, 15614966, 15905500, 16031731, 16259687, 16302225, 16306449, 16545046, 16577832, 16708904),
+            'pin28': (2100167, 4736763, 13920936, 16272063),
+            'pin32': (1830, 1073899, 1097544, 1366761, 1859371, 2927397, 3179945, 3179945, 4779532, 5260893, 5506214, 8396546, 8398473, 9473400, 13131776, 14221027, 15256600, 16018692, 16550807),
+            'pinDLink': (5329, 1365581, 1867493, 2625659, 8702386, 10529563, 12100486, 12624059, 13414997, 14216087, 16545046),
+            'pinDLink1': (5329, 6375, 6491, 7408, 7768, 8593, 8880, 9217, 9818, 1365581, 1867493, 3409924, 6085016, 8702386, 12100486, 13155865, 13161379, 13414997),
+            'pinASUS': (1830, 1830, 6012, 6012, 7846, 12367, 57420, 298296, 299558, 528503, 528504, 528505, 540253, 548974, 549478, 1080132, 1097544, 1098619, 1113837, 1367465, 1580664, 1852441, 1869612, 1881900, 2367687, 2391840, 2905820, 2927397, 2948513, 3168826, 3179945, 3681354, 3724615, 3939844, 4200062, 4256257, 4516317, 4779532, 5260893, 5530841, 5546064, 5552138, 5798889, 6309323, 6333516, 6345130, 6574462, 6609236, 7084431, 7107104, 7142841, 7359867, 7365304, 7655467, 7873711, 7885870, 7920031, 8136292, 8404829, 8692771, 8955590, 8968182, 9179348, 9209899, 9456970, 9466498, 9500242, 9763762, 10247310, 10492713, 10548161, 11073504, 11280907, 11312663, 11313683, 11562687, 12080400, 12119566, 12334080, 12359296, 12381819, 12624059, 12849909, 12888093, 13131776, 13144569, 13635289, 13637570, 13665456, 14176486, 14221027, 14696265, 14991085, 15242486, 15256600, 15473241, 15475328, 15486029, 15759705, 16001107, 16006753, 16018415, 16265956, 16296709, 16312579, 16550807),
+            'pinAirocon': (1830, 2859, 3828, 4915, 6012, 6895, 57419, 135192, 528499, 528503, 1053678, 2927397, 7900244, 8404829, 9763762, 12359296, 16006753, 16550807),
+            'pinEmpty': (3727, 19063, 825023, 1073899, 1097461, 1859371, 2159523, 2921855, 3427818, 3725359, 3971186, 5553747, 5821806, 6558572, 6856950, 7351842, 7380781, 7645070, 7648638, 7658202, 7897346, 7902388, 7902850, 8141139, 8398473, 8966772, 9201864, 9742263, 9966387, 10278467, 10529563, 11330069, 13160798, 13280102, 13656204, 13902114, 13918435, 13924074, 14704742, 14970643, 15475328),
+            'pinCisco': (6699, 9356, 9752, 3427819, 7369148, 14707093, 14732110),
+            'pinBrcm1': (6825, 1315915, 9997149, 11334111, 12383877, 13161379, 15491684),
+            'pinBrcm2': (1365581, 1867493, 2625659, 8702386, 12100486, 12383877, 13155865),
+            'pinBrcm3': (1365581, 1867493, 2625659, 8127308, 12100486, 12383877, 13155865),
+            'pinBrcm4': (1365581, 1597996, 1867493, 2117247, 2625659, 4986859, 8127448, 8702386, 12100486, 12383877, 13155865, 13161379, 13414997, 14183657, 16545046),
+            'pinBrcm5': (1365581, 1597996, 1867493, 2117247, 2625659, 4986859, 8127448, 8702386, 12100486, 12383877, 13155865, 13161379, 13414997, 14183657, 16545046),
+            'pinBrcm6': (1365581, 1597996, 1867493, 2117247, 2625659, 4986859, 8127448, 8702386, 12100486, 12383877, 13155865, 13161379, 13414997, 14183657, 16545046),
+            'pinAirc1': (1580664, 4256257, 4516317, 13665456),
+            'pinAirc2': (8692771, 8955590, 9179348),
+            'pinDSL2740R': (9818, 1883577, 3409924, 6085016, 8702386, 16545046),
+            'pinRealtek1': (3138, 3816, 5329),
+            'pinRealtek2': (29283, 14991085),
+            'pinRealtek3': (575155,),
+            'pinUpvel': (16302225,),
+            'pinUR814AC': (13942655,),
+            'pinUR825AC': (13942655,),
+            'pinOnlime': (5329, 7881846, 13942655, 16302225),
+            'pinEdimax': (57420, 8396546),
+            'pinThomson': (9764, 4469448, 8976327, 13370362),
+            'pinHG532x': (26187, 549729, 555596, 825023, 1358184, 2099437, 2386341, 3435475, 7891593, 8971179, 10273138, 11330069, 13410851, 13662901, 15256877, 16253203, 16268799),
+            'pinH108L': (4983220, 5024778, 10277451, 11564501, 13132999, 14418574, 16566423),
+            'pinONO': (6042939, 14439292)
+        }
+        res = []
+        for ID, OUI in algorithms.items():
+            if oui in OUI:
+                res.append(ID)
+        return res
+
     def pin24(self, mac):
         return (mac & 0xFFFFFF)