Fix dividing out direction of arrow

yuuki · Mar 4, 2018 · ef757fb · ef757fb
1 parent 29111aa
commit ef757fb
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/conntrack/conntrack.go b/conntrack/conntrack.go
@@ -100,7 +100,13 @@ func (f *HostFlow) HasDirection(dire FlowDirection) bool {
 
 // String returns the string representation of HostFlow.
 func (f *HostFlow) String() string {
-	return fmt.Sprintf("%s\t --> \t%s \t%s", f.Local, f.Peer, f.Stat)
+	switch f.Direction {
+	case FlowActive:
+		return fmt.Sprintf("%s\t --> \t%s \t%s", f.Local, f.Peer, f.Stat)
+	case FlowPassive:
+		return fmt.Sprintf("%s\t <-- \t%s \t%s", f.Local, f.Peer, f.Stat)
+	}
+	return ""
 }
 
 // ReplaceLookupedName replaces f.Addr into lookuped name.