github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-16 10:03:39 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: do not emit graph if empty

Browse source
This commit is contained in:
Simone Margaritelli 2021-04-09 12:45:06 +02:00
parent 1d306e6cd2
commit 0042b77c36
2 changed files with 33 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

27
modules/graph/dot.go
View file

@ -23,18 +23,23 @@ func (mod *Module) generateDotGraph(bssid string) error {
return err return err
} }
if mod.settings.privacy { if size > 0 {
data = privacyFilter.ReplaceAllString(data, "$1:$2:xx:xx:xx:xx") if mod.settings.privacy {
data = privacyFilter.ReplaceAllString(data, "$1:$2:xx:xx:xx:xx")
}
if err := ioutil.WriteFile(mod.settings.dot.output, []byte(data), os.ModePerm); err != nil {
return err
} else {
mod.Info("graph saved to %s in %v (%d edges, %d discarded)",
mod.settings.dot.output,
time.Since(start),
size,
discarded)
}
} else {
mod.Info("graph is empty")
} }
if err := ioutil.WriteFile(mod.settings.dot.output, []byte(data), os.ModePerm); err != nil {
return err
} else {
mod.Info("graph saved to %s in %v (%d edges, %d discarded)",
mod.settings.dot.output,
time.Since(start),
size,
discarded)
}
return nil return nil
} }

28
modules/graph/json.go
View file

@ -20,18 +20,24 @@ func (mod *Module) generateJSONGraph(bssid string) error {
return err return err
} }
if mod.settings.privacy { if size > 0 {
data = privacyFilter.ReplaceAllString(data, "$1:$2:xx:xx:xx:xx")
if mod.settings.privacy {
data = privacyFilter.ReplaceAllString(data, "$1:$2:xx:xx:xx:xx")
}
if err := ioutil.WriteFile(mod.settings.json.output, []byte(data), os.ModePerm); err != nil {
return err
} else {
mod.Info("graph saved to %s in %v (%d edges, %d discarded)",
mod.settings.json.output,
time.Since(start),
size,
discarded)
}
} else {
mod.Info("graph is empty")
} }
if err := ioutil.WriteFile(mod.settings.json.output, []byte(data), os.ModePerm); err != nil {
return err
} else {
mod.Info("graph saved to %s in %v (%d edges, %d discarded)",
mod.settings.json.output,
time.Since(start),
size,
discarded)
}
return nil return nil
} }