refact: refactored net.show in a separate file

This commit is contained in:
evilsocket 2018-01-13 20:34:36 +01:00
commit 3e15e13027
2 changed files with 44 additions and 38 deletions

View file

@ -1,11 +1,8 @@
package session
import (
"fmt"
"sort"
"sync"
"github.com/evilsocket/bettercap-ng/core"
"github.com/evilsocket/bettercap-ng/net"
)
@ -77,38 +74,3 @@ func (tp *Targets) AddIfNotExist(ip, mac string) *net.Endpoint {
return nil
}
type tSorter []*net.Endpoint
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 (tp *Targets) Dump() {
tp.lock.Lock()
defer tp.lock.Unlock()
fmt.Println()
fmt.Printf(" " + core.GREEN + "interface" + core.RESET + "\n\n")
fmt.Printf(" " + tp.Interface.String() + "\n")
fmt.Println()
fmt.Printf(" " + core.GREEN + "gateway" + core.RESET + "\n\n")
fmt.Printf(" " + tp.Gateway.String() + "\n")
if len(tp.Targets) > 0 {
fmt.Println()
fmt.Printf(" " + core.GREEN + "hosts" + core.RESET + "\n\n")
targets := make([]*net.Endpoint, 0, len(tp.Targets))
for _, t := range tp.Targets {
targets = append(targets, t)
}
sort.Sort(tSorter(targets))
for _, t := range targets {
fmt.Println(" " + t.String())
}
}
fmt.Println()
}