From 02c8a095d3e09a3eb126bbcb3ab750a5c3e55b37 Mon Sep 17 00:00:00 2001
From: evilsocket <evilsocket@gmail.com>
Date: Tue, 16 Jan 2018 10:24:24 +0100
Subject: [PATCH] fix: sorting protocol hits in net.show command output

---
 modules/net_recon.go | 27 +++++++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

diff --git a/modules/net_recon.go b/modules/net_recon.go
index 6ec44e57..5410fd22 100644
--- a/modules/net_recon.go
+++ b/modules/net_recon.go
@@ -159,6 +159,28 @@ func (a tSorter) Len() int           { return len(a) }
 func (a tSorter) Swap(i, j int)      { a[i], a[j] = a[j], a[i] }
 func (a tSorter) Less(i, j int) bool { return a[i].IpAddressUint32 < a[j].IpAddressUint32 }
 
+func rankByProtoHits(protos map[string]uint64) ProtoPairList {
+	pl := make(ProtoPairList, len(protos))
+	i := 0
+	for k, v := range protos {
+		pl[i] = ProtoPair{k, v}
+		i++
+	}
+	sort.Sort(sort.Reverse(pl))
+	return pl
+}
+
+type ProtoPair struct {
+	Protocol string
+	Hits     uint64
+}
+
+type ProtoPairList []ProtoPair
+
+func (p ProtoPairList) Len() int           { return len(p) }
+func (p ProtoPairList) Less(i, j int) bool { return p[i].Hits < p[j].Hits }
+func (p ProtoPairList) Swap(i, j int)      { p[i], p[j] = p[j], p[i] }
+
 func (d *Discovery) Show() error {
 	d.Session.Targets.Lock()
 	defer d.Session.Targets.Unlock()
@@ -235,8 +257,9 @@ func (d *Discovery) Show() error {
 
 	table = tablewriter.NewWriter(os.Stdout)
 
-	for proto, hits := range d.Session.Queue.Protos {
-		table.Append([]string{proto, fmt.Sprintf("%d", hits)})
+	protos := rankByProtoHits(d.Session.Queue.Protos)
+	for _, p := range protos {
+		table.Append([]string{p.Protocol, fmt.Sprintf("%d", p.Hits)})
 	}
 
 	table.SetHeader([]string{"Proto", "# Packets"})