87 lines
3.0 KiB
Python
Executable File
87 lines
3.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from ..model.attack import Attack
|
|
from ..util.color import Color
|
|
from ..util.process import Process
|
|
from ..config import Configuration
|
|
from ..tools.bully import Bully
|
|
from ..tools.reaver import Reaver
|
|
|
|
class AttackWPS(Attack):
|
|
|
|
@staticmethod
|
|
def can_attack_wps():
|
|
return Reaver.exists() or Bully.exists()
|
|
|
|
def __init__(self, target, pixie_dust=False):
|
|
super(AttackWPS, self).__init__(target)
|
|
self.success = False
|
|
self.crack_result = None
|
|
self.pixie_dust = pixie_dust
|
|
|
|
def run(self):
|
|
''' Run all WPS-related attacks '''
|
|
|
|
# Drop out if user specified to not use Reaver/Bully
|
|
if Configuration.use_pmkid_only:
|
|
self.success = False
|
|
return False
|
|
|
|
if Configuration.no_wps:
|
|
self.success = False
|
|
return False
|
|
|
|
if not Configuration.wps_pixie and self.pixie_dust:
|
|
Color.pl('\r{!} {O}--no-pixie{R} was given, ignoring WPS PIN Attack on ' +
|
|
'{O}%s{W}' % self.target.essid)
|
|
self.success = False
|
|
return False
|
|
|
|
if not Configuration.wps_pin and not self.pixie_dust:
|
|
Color.pl('\r{!} {O}--no-pin{R} was given, ignoring WPS Pixie-Dust Attack ' +
|
|
'on {O}%s{W}' % self.target.essid)
|
|
self.success = False
|
|
return False
|
|
|
|
if not Reaver.exists() and Bully.exists():
|
|
# Use bully if reaver isn't available
|
|
return self.run_bully()
|
|
elif self.pixie_dust and not Reaver.is_pixiedust_supported() and Bully.exists():
|
|
# Use bully if reaver can't do pixie-dust
|
|
return self.run_bully()
|
|
elif Configuration.use_bully:
|
|
# Use bully if asked by user
|
|
return self.run_bully()
|
|
elif not Reaver.exists():
|
|
# Print error if reaver isn't found (bully not available)
|
|
if self.pixie_dust:
|
|
Color.pl('\r{!} {R}Skipping WPS Pixie-Dust attack: {O}reaver{R} not found.{W}')
|
|
else:
|
|
Color.pl('\r{!} {R}Skipping WPS PIN attack: {O}reaver{R} not found.{W}')
|
|
return False
|
|
elif self.pixie_dust and not Reaver.is_pixiedust_supported():
|
|
# Print error if reaver can't support pixie-dust (bully not available)
|
|
Color.pl('\r{!} {R}Skipping WPS attack: {O}reaver{R} does not support {O}--pixie-dust{W}')
|
|
return False
|
|
else:
|
|
return self.run_reaver()
|
|
|
|
|
|
def run_bully(self):
|
|
bully = Bully(self.target, pixie_dust=self.pixie_dust)
|
|
bully.run()
|
|
bully.stop()
|
|
self.crack_result = bully.crack_result
|
|
self.success = self.crack_result is not None
|
|
return self.success
|
|
|
|
|
|
def run_reaver(self):
|
|
reaver = Reaver(self.target, pixie_dust=self.pixie_dust)
|
|
reaver.run()
|
|
self.crack_result = reaver.crack_result
|
|
self.success = self.crack_result is not None
|
|
return self.success
|
|
|