github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-19 13:09:49 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

balls

Browse source
This commit is contained in:
evilsocket 2018-01-16 10:43:59 +01:00
commit 4b9cd477aa
1 changed files with 20 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

24
modules/net_recon.go
View file

@ -159,15 +159,19 @@ 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) 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 (a tSorter) Less(i, j int) bool { return a[i].IpAddressUint32 < a[j].IpAddressUint32 }
func rankByProtoHits(protos map[string]uint64) ProtoPairList { func rankByProtoHits(protos map[string]uint64) (ProtoPairList, uint64) {
pl := make(ProtoPairList, len(protos)) pl := make(ProtoPairList, len(protos))
max := uint64(0)
i := 0 i := 0
for k, v := range protos { for k, v := range protos {
pl[i] = ProtoPair{k, v} pl[i] = ProtoPair{k, v}
if v > max {
max = v
}
i++ i++
} }
sort.Sort(sort.Reverse(pl)) sort.Sort(sort.Reverse(pl))
return pl return pl, max
} }
type ProtoPair struct { type ProtoPair struct {
@ -197,6 +201,7 @@ func (d *Discovery) Show() error {
table := tablewriter.NewWriter(os.Stdout) table := tablewriter.NewWriter(os.Stdout)
table.SetColWidth(80)
table.AppendBulk(data) table.AppendBulk(data)
table.Render() table.Render()
@ -236,6 +241,7 @@ func (d *Discovery) Show() error {
table = tablewriter.NewWriter(os.Stdout) table = tablewriter.NewWriter(os.Stdout)
table.SetHeader([]string{"IP", "MAC", "Hostname", "Vendor", "Sent", "Recvd", "Last Seen"}) table.SetHeader([]string{"IP", "MAC", "Hostname", "Vendor", "Sent", "Recvd", "Last Seen"})
table.SetColWidth(80)
table.AppendBulk(data) table.AppendBulk(data)
table.Render() table.Render()
@ -252,16 +258,26 @@ func (d *Discovery) Show() error {
table = tablewriter.NewWriter(os.Stdout) table = tablewriter.NewWriter(os.Stdout)
table.SetHeader([]string{"Sent", "Sniffed", "# Packets", "Errors"}) table.SetHeader([]string{"Sent", "Sniffed", "# Packets", "Errors"})
table.SetColWidth(80)
table.Append(row) table.Append(row)
table.Render() table.Render()
fmt.Println() fmt.Println()
table = tablewriter.NewWriter(os.Stdout) table = tablewriter.NewWriter(os.Stdout)
table.SetColWidth(80)
protos, maxPackets := rankByProtoHits(d.Session.Queue.Protos)
maxBarWidth := 70
protos := rankByProtoHits(d.Session.Queue.Protos)
for _, p := range protos { for _, p := range protos {
table.Append([]string{p.Protocol, fmt.Sprintf("%d", p.Hits)}) width := int(float32(maxBarWidth) * (float32(p.Hits) / float32(maxPackets)))
bar := ""
for i := 0; i < width; i++ {
bar += "▇"
}
table.Append([]string{p.Protocol, fmt.Sprintf("%s %d", bar, p.Hits)})
} }
table.SetHeader([]string{"Proto", "# Packets"}) table.SetHeader([]string{"Proto", "# Packets"})