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

fix: save unsaved packets only if handshake is complete

Browse source
This commit is contained in:
evilsocket 2019-02-06 14:07:58 +01:00
commit e9b93b8586
No known key found for this signature in database
GPG key ID: 1564D7F30393A456
1 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
network/wifi.go
View file

@ -220,11 +220,16 @@ func (w *WiFi) SaveHandshakesTo(fileName string, linkType layers.LinkType) error
for _, ap := range w.aps {
for _, station := range ap.Clients() {
station.Handshake.EachUnsavedPacket(func(pkt gopacket.Packet) {
err = writer.WritePacket(pkt.Metadata().CaptureInfo, pkt.Data())
})
if err != nil {
return err
if station.Handshake.Complete() {
err = nil
station.Handshake.EachUnsavedPacket(func(pkt gopacket.Packet) {
if err == nil {
err = writer.WritePacket(pkt.Metadata().CaptureInfo, pkt.Data())
}
})
if err != nil {
return err
}
}
}
}