#!/usr/bin/python2.7
# -*- coding: utf-8 -*-

from Attack import Attack
from Color import Color
from Configuration import Configuration
from Bully import Bully
from Reaver import Reaver

class AttackWPS(Attack):
    def __init__(self, target):
        super(AttackWPS, self).__init__(target)
        self.success = False
        self.crack_result = None

    def run(self):
        ''' Run all WPS-related attacks '''

        # Drop out if user specified to not use Reaver/Bully
        if Configuration.no_wps:
            Color.pl('\r{!} {O}--no-wps{R} set, ignoring WPS attack on {O}%s{W}' % self.target.essid)
            self.success = False
            return self.success

        ###################
        # Pixie-Dust attack
        if Configuration.use_bully:
            # Bully: Pixie-dust
            bully = Bully(self.target)
            bully.run()
            bully.stop()
            if bully.crack_result is not None:
                self.crack_result = bully.crack_result
                self.success = True
                return True
        else:
            reaver = Reaver(self.target)
            if reaver.is_pixiedust_supported():
                # Reaver: Pixie-dust
                reaver = Reaver(self.target)
                if reaver.run_pixiedust_attack():
                    self.success = True
                    return True
            else:
                Color.pl("{!} {R}your version of 'reaver' does not support the {O}WPS pixie-dust attack{W}")

        return False