github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-20 05:23:19 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: collect additional frames for stations with key material (ref #810)

Browse source
This commit is contained in:
Simone Margaritelli 2021-03-30 23:19:06 +02:00
commit cea53b969e
3 changed files with 73 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
network/wifi_handshake.go
View file

@ -79,6 +79,12 @@ func (h *Handshake) AddFrame(n int, pkt gopacket.Packet) {
h.unsaved = append(h.unsaved, pkt)
}
func (h *Handshake) AddExtra(pkt gopacket.Packet) {
h.Lock()
defer h.Unlock()
h.unsaved = append(h.unsaved, pkt)
}
func (h *Handshake) Complete() bool {
h.RLock()
defer h.RUnlock()
@ -115,6 +121,10 @@ func (h *Handshake) HasPMKID() bool {
return h.hasPMKID
}
func (h *Handshake) Any() bool {
return h.HasPMKID() || h.Half() || h.Complete()
}
func (h *Handshake) NumUnsaved() int {
h.RLock()
defer h.RUnlock()