github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-15 01:23:42 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

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

Browse source
This commit is contained in:
evilsocket 2018-02-19 20:58:10 +01:00
parent 420ab9e26c
commit b4df469ff1
2 changed files with 16 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/wifi_recon.go
View file

@ -311,7 +311,7 @@ func (w *WiFiRecon) startDeauth(apMac net.HardwareAddr, clMac net.HardwareAddr)
} else {
log.Info("Deauthing clients from AP %s ...", apMac.String())
// deauth every authenticated client
for _, station := range w.Session.WiFi.Stations {
for _, station := range w.Session.WiFi.List() {
if station.IsAP == false {
w.sendDeauthPacket(apMac, station.HW)
}

31
network/wifi.go
View file

@ -10,19 +10,18 @@ type StationLostCallback func(s *Station)
type WiFi struct {
sync.Mutex
Interface *Endpoint
Stations map[string]*Station
newCb StationNewCallback
lostCb StationLostCallback
iface *Endpoint
stations map[string]*Station
newCb StationNewCallback
lostCb StationLostCallback
}
func NewWiFi(iface *Endpoint, newcb StationNewCallback, lostcb StationLostCallback) *WiFi {
return &WiFi{
Interface: iface,
Stations: make(map[string]*Station),
newCb: newcb,
lostCb: lostcb,
iface: iface,
stations: make(map[string]*Station),
newCb: newcb,
lostCb: lostcb,
}
}
@ -31,7 +30,7 @@ func (w *WiFi) List() (list []*Station) {
defer w.Unlock()
list = make([]*Station, 0)
for _, t := range w.Stations {
for _, t := range w.stations {
list = append(list, t)
}
return
@ -41,8 +40,8 @@ func (w *WiFi) Remove(mac string) {
w.Lock()
defer w.Unlock()
if s, found := w.Stations[mac]; found {
delete(w.Stations, mac)
if s, found := w.stations[mac]; found {
delete(w.stations, mac)
if w.lostCb != nil {
w.lostCb(s)
}
@ -54,14 +53,14 @@ func (w *WiFi) AddIfNew(ssid, mac string, isAp bool, channel int, rssi int8) *St
defer w.Unlock()
mac = NormalizeMac(mac)
if station, found := w.Stations[mac]; found {
if station, found := w.stations[mac]; found {
station.LastSeen = time.Now()
station.RSSI = rssi
return station
}
newStation := NewStation(ssid, mac, isAp, channel, rssi)
w.Stations[mac] = newStation
w.stations[mac] = newStation
if w.newCb != nil {
w.newCb(newStation)
@ -75,11 +74,11 @@ func (w *WiFi) Get(mac string) (*Station, bool) {
defer w.Unlock()
mac = NormalizeMac(mac)
station, found := w.Stations[mac]
station, found := w.stations[mac]
return station, found
}
func (w *WiFi) Clear() error {
w.Stations = make(map[string]*Station)
w.stations = make(map[string]*Station)
return nil
}