fix: check ssid probes for non printable characters

bettercap · Apr 7, 2021 · 5b8cb9a · 5b8cb9a
1 parent db27542
commit 5b8cb9a
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/modules/graph/node.go b/modules/graph/node.go
@@ -3,6 +3,7 @@ package graph
 import (
  "fmt"
  "time"
+ "unicode"
 )
 
 type NodeType string
@@ -50,7 +51,20 @@ func (n Node) String() string {
 func (n Node) Label() string {
  switch n.Type {
  case SSID:
- return n.Entity.(string)
+ s := n.Entity.(string)
+ allPrint := true
+
+ for _, rn := range s {
+ if !unicode.IsPrint(rune(rn)) {
+ allPrint = false
+ break
+ }
+ }
+
+ if !allPrint {
+ s = fmt.Sprintf("0x%x", s)
+ }
+ return s
  case BLEServer:
  return fmt.Sprintf("%s\\n(%s)",
  n.Entity.(map[string]interface{})["mac"].(string),