github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-08 05:51:37 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: gateway cannot found on macos since netstat output is different from linux

Browse source
This commit is contained in:
Onur 2018-01-18 14:24:03 +03:00
parent 97c97a668a
commit c7aff6baa0
1 changed files with 25 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

31
net/net.go
View file

@ -4,6 +4,7 @@ import (
"fmt" "fmt"
"net" "net"
"regexp" "regexp"
"runtime"
"strconv" "strconv"
"strings" "strings"
@ -12,6 +13,10 @@ import (
var IPv4RouteParser = regexp.MustCompile("^([\\d\\.]+)\\s+([\\d\\.]+)\\s+([\\d\\.]+)\\s+([A-Z]+)\\s+\\d+\\s+\\d+\\s+\\d+\\s+(.+)$") var IPv4RouteParser = regexp.MustCompile("^([\\d\\.]+)\\s+([\\d\\.]+)\\s+([\\d\\.]+)\\s+([A-Z]+)\\s+\\d+\\s+\\d+\\s+\\d+\\s+(.+)$")
var IPv4RouteTokens = 6 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) { func FindInterface(name string) (*Endpoint, error) {
ifaces, err := net.Interfaces() ifaces, err := net.Interfaces()
@ -70,20 +75,34 @@ func FindInterface(name string) (*Endpoint, error) {
} }
func FindGateway(iface *Endpoint) (*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("netstat", []string{"-n", "-r"})
if err != nil { if err != nil {
return nil, err return nil, err
} }
for _, line := range strings.Split(output, "\n") { for _, line := range strings.Split(output, "\n") {
m := IPv4RouteParser.FindStringSubmatch(line) m := routeParser.FindStringSubmatch(line)
if len(m) == IPv4RouteTokens { if len(m) == routeTokens {
// destination := m[1] // destination := m[1]
// mask := m[3] // mask := m[3]
flags := m[4] flags := m[flagsIndex]
ifname := m[5] ifname := m[ifnameIndex]
if ifname == iface.Name() && flags == gwFlags {
if ifname == iface.Name() && flags == "UG" {
gateway := m[2] gateway := m[2]
if gateway == iface.IpAddress { if gateway == iface.IpAddress {
return iface, nil return iface, nil