github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-14 02:36:57 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Replace netstat from net-tools by ip from iproute2 (linux)

Browse source
This commit is contained in:
Daniel Rodriguez 2018-01-20 21:11:17 +01:00
commit 881126941b
3 changed files with 45 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

38
net/net.go
View file

@ -4,20 +4,12 @@ import (
"fmt"
"net"
"regexp"
"runtime"
"strconv"
"strings"
"github.com/evilsocket/bettercap-ng/core"
)
var IPv4RouteParser = regexp.MustCompile("^([\\d\\.]+)\\s+([\\d\\.]+)\\s+([\\d\\.]+)\\s+([A-Z]+)\\s+\\d+\\s+\\d+\\s+\\d+\\s+(.+)$")
var IPv4RouteTokens = 6
var IPv4RouteParserMac = regexp.MustCompile("^([a-z]+)+\\s+(\\d+\\.+\\d+.\\d.+\\d)+\\s+([a-zA-z]+)+\\s+(\\d+)+\\s+(\\d+)+\\s+([a-zA-Z]+\\d+)$")
var IPv4RouteTokensMac = 7
var IPv4RouteGWFlags = "UG"
var IPv4RouteGWFlagsMac = "UGSc"
func FindInterface(name string) (*Endpoint, error) {
ifaces, err := net.Interfaces()
if err != nil {
@ -75,35 +67,15 @@ func FindInterface(name string) (*Endpoint, error) {
}
func FindGateway(iface *Endpoint) (*Endpoint, error) {
var routeParser = IPv4RouteParser
var routeTokens = IPv4RouteTokens
var flagsIndex = 4
var ifnameIndex = 5
var gwFlags = IPv4RouteGWFlags
if runtime.GOOS == "darwin" {
// "MacOS detected"
routeParser = IPv4RouteParserMac
routeTokens = IPv4RouteTokensMac
flagsIndex = 3
ifnameIndex = 6
gwFlags = IPv4RouteGWFlagsMac
}
output, err := core.Exec("netstat", []string{"-n", "-r"})
output, err := core.Exec(IPv4RouteCmd, IPv4RouteCmdOpts)
if err != nil {
return nil, err
}
for _, line := range strings.Split(output, "\n") {
m := routeParser.FindStringSubmatch(line)
if len(m) == routeTokens {
// destination := m[1]
// mask := m[3]
flags := m[flagsIndex]
ifname := m[ifnameIndex]
if ifname == iface.Name() && flags == gwFlags {
gateway := m[2]
m := IPv4RouteParser.FindStringSubmatch(line)
if len(m) == IPv4RouteTokens {
return IPv4RouteIsGateway(iface.Name(), m, func(gateway string) (*Endpoint, error) {
if gateway == iface.IpAddress {
return iface, nil
} else {
@ -114,7 +86,7 @@ func FindGateway(iface *Endpoint) (*Endpoint, error) {
}
return NewEndpoint(gateway, mac), nil
}
}
})
}
}