misc: small fix or general refactoring i did not bother commenting

This commit is contained in:
evilsocket 2018-02-16 13:48:55 +01:00
parent c2467ac7a4
commit b48c75a7fc

View file

@ -256,21 +256,23 @@ func (w *WDiscovery) sendDeauthPacket(ap net.HardwareAddr, client net.HardwareAd
} }
func (w *WDiscovery) startDeauth() error { func (w *WDiscovery) startDeauth() error {
switch { isTargetingAP := len(w.apTarget) > 0
case len(w.apTarget) > 0 && len(w.cliTarget) > 0: isTargetingCLI := len(w.cliTarget) > 0
w.sendDeauthPacket(w.apTarget, w.cliTarget)
case len(w.apTarget) > 0: if isTargetingAP {
if isTargetingCLI {
// deauth a specific client
w.sendDeauthPacket(w.apTarget, w.cliTarget)
} else {
// deauth all AP's clients
for _, t := range w.Stations.Stations { for _, t := range w.Stations.Stations {
w.sendDeauthPacket(w.apTarget, t.Endpoint.HW) w.sendDeauthPacket(w.apTarget, t.Endpoint.HW)
} }
default:
return errors.New("Base station is not set.")
} }
return nil return nil
} }
return errors.New("No base station or client set.")
}
func (w *WDiscovery) discoverAccessPoints(packet gopacket.Packet) { func (w *WDiscovery) discoverAccessPoints(packet gopacket.Packet) {
var bssid string var bssid string