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

This commit is contained in:
evilsocket 2019-08-22 13:48:52 -04:00
commit ff3add0fe2
No known key found for this signature in database
GPG key ID: 82E42E7F3B34C97E
3 changed files with 9 additions and 6 deletions

View file

@ -364,14 +364,14 @@ func (mod *WiFiModule) Configure() error {
return err
}
if err, mod.shakesFile = mod.StringParam("wifi.handshakes.file"); err != nil {
if err, mod.shakesAggregate = mod.BoolParam("wifi.handshakes.aggregate"); err != nil {
return err
} else if err, mod.shakesFile = mod.StringParam("wifi.handshakes.file"); err != nil {
return err
} else if mod.shakesFile != "" {
if mod.shakesFile, err = fs.Expand(mod.shakesFile); err != nil {
return err
}
} else if err, mod.shakesAggregate = mod.BoolParam("wifi.handshakes.aggregate"); err != nil {
return err
}
if err, ifName = mod.StringParam("wifi.interface"); err != nil {

View file

@ -89,7 +89,7 @@ func (mod *WiFiModule) discoverHandshakes(radiotap *layers.RadioTap, dot11 *laye
numUnsaved := station.Handshake.NumUnsaved()
shakesFileName := mod.shakesFile
if mod.shakesAggregate == false {
shakesFileName = path.Join(shakesFileName, fmt.Sprintf("%s.pcap", station.PathFriendlyName()))
shakesFileName = path.Join(shakesFileName, fmt.Sprintf("%s.pcap", ap.PathFriendlyName()))
}
doSave := numUnsaved > 0
if doSave && shakesFileName != "" {