diff --git a/modules/graph/dot.go b/modules/graph/dot.go index 83c29e0b..e6f2b491 100644 --- a/modules/graph/dot.go +++ b/modules/graph/dot.go @@ -23,18 +23,23 @@ func (mod *Module) generateDotGraph(bssid string) error { return err } - if mod.settings.privacy { - data = privacyFilter.ReplaceAllString(data, "$1:$2:xx:xx:xx:xx") + if size > 0 { + 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 } \ No newline at end of file diff --git a/modules/graph/json.go b/modules/graph/json.go index 90e34f73..dccd725a 100644 --- a/modules/graph/json.go +++ b/modules/graph/json.go @@ -20,18 +20,24 @@ func (mod *Module) generateJSONGraph(bssid string) error { return err } - if mod.settings.privacy { - data = privacyFilter.ReplaceAllString(data, "$1:$2:xx:xx:xx:xx") + if size > 0 { + + 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 } \ No newline at end of file