github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-14 02:36:57 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
evilsocket 2018-02-20 19:03:45 +01:00
commit 51f563d830
2 changed files with 9 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

3
modules/wifi_recon.go
View file

@ -6,7 +6,6 @@ import (
"os"
"sort"
"strconv"
"strings"
"time"
"github.com/evilsocket/bettercap-ng/core"
@ -415,7 +414,7 @@ func (w *WiFiRecon) updateStats(dot11 *layers.Dot11, packet gopacket.Packet) {
if ok, enc := packets.Dot11ParseEncryption(packet, dot11); ok == true {
bssid := dot11.Address3.String()
if station, found := w.Session.WiFi.Get(bssid); found == true {
station.Encryption = strings.Join(enc, ", ")
station.Encryption = enc
}
}
}

16
packets/dot11.go
View file

@ -64,13 +64,13 @@ func Dot11ParseIDSSID(packet gopacket.Packet) (bool, string) {
return false, ""
}
func Dot11ParseEncryption(packet gopacket.Packet, dot11 *layers.Dot11) (bool, []string) {
enc := make([]string, 0)
func Dot11ParseEncryption(packet gopacket.Packet, dot11 *layers.Dot11) (bool, string) {
enc := ""
found := false
if dot11.Flags.WEP() {
found = true
enc = append(enc, "WEP")
enc = "WEP"
}
for _, layer := range packet.Layers() {
@ -79,16 +79,16 @@ func Dot11ParseEncryption(packet gopacket.Packet, dot11 *layers.Dot11) (bool, []
if ok == true {
found = true
if info.ID == layers.Dot11InformationElementIDRSNInfo {
enc = append(enc, "WPA2")
} else if info.ID == layers.Dot11InformationElementIDVendor && bytes.Index(info.Info, []byte{0, 0x50, 0xf2, 1, 1, 0}) == 0 {
enc = append(enc, "WPA")
enc = "WPA2"
} else if info.ID == layers.Dot11InformationElementIDVendor && bytes.Index(info.OUI, []byte{0, 0x50, 0xf2, 1, 1, 0}) == 0 {
enc = "WPA"
}
}
}
}
if found && len(enc) == 0 {
enc = append(enc, "OPEN")
if found && enc == "" {
enc = "OPEN"
}
return found, enc