github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-08 05:51:37 -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-17 01:57:14 +01:00
parent d775dd3b3c
commit e1e2e33f57
3 changed files with 87 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

36
modules/wifi_stats.go
View file

@ -6,7 +6,8 @@ import (
)
type WiFiStationStats struct {
Bytes uint64
Sent uint64
Received uint64
}
type WiFiStats struct {
@ -20,25 +21,48 @@ func NewWiFiStats() *WiFiStats {
}
}
func (s *WiFiStats) Collect(station net.HardwareAddr, bytes uint64) {
func (s *WiFiStats) CollectSent(station net.HardwareAddr, bytes uint64) {
s.Lock()
defer s.Unlock()
bssid := station.String()
if sstats, found := s.stats[bssid]; found == true {
sstats.Bytes += bytes
sstats.Sent += bytes
} else {
s.stats[bssid] = &WiFiStationStats{Bytes: bytes}
s.stats[bssid] = &WiFiStationStats{Sent: bytes}
}
}
func (s *WiFiStats) For(station net.HardwareAddr) uint64 {
func (s *WiFiStats) CollectReceived(station net.HardwareAddr, bytes uint64) {
s.Lock()
defer s.Unlock()
bssid := station.String()
if sstats, found := s.stats[bssid]; found == true {
return sstats.Bytes
sstats.Received += bytes
} else {
s.stats[bssid] = &WiFiStationStats{Received: bytes}
}
}
func (s *WiFiStats) SentFrom(station net.HardwareAddr) uint64 {
s.Lock()
defer s.Unlock()
bssid := station.String()
if sstats, found := s.stats[bssid]; found == true {
return sstats.Sent
}
return 0
}
func (s *WiFiStats) SentTo(station net.HardwareAddr) uint64 {
s.Lock()
defer s.Unlock()
bssid := station.String()
if sstats, found := s.stats[bssid]; found == true {
return sstats.Received
}
return 0
}