gateway
This commit is contained in:
crusader
parent
5db26291ed
commit
7c6cac9569
131
net/gateway/gateway_common.go
Normal file
131
net/gateway/gateway_common.go
Normal file
|
|
@ -0,0 +1,131 @@
|
|||
package gateway
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net"
|
||||
"strings"
|
||||
)
|
||||
|
||||
var errNoGateway = errors.New("no gateway found")
|
||||
|
||||
func parseWindowsRoutePrint(output []byte) (net.IP, string, error) {
|
||||
// Windows route output format is always like this:
|
||||
// ===========================================================================
|
||||
// Active Routes:
|
||||
// Network Destination Netmask Gateway Interface Metric
|
||||
// 0.0.0.0 0.0.0.0 192.168.1.1 192.168.1.100 20
|
||||
// ===========================================================================
|
||||
// I'm trying to pick the active route,
|
||||
// then jump 2 lines and pick the third IP
|
||||
// Not using regex because output is quite standard from Windows XP to 8 (NEEDS TESTING)
|
||||
lines := strings.Split(string(output), "\n")
|
||||
for idx, line := range lines {
|
||||
if strings.HasPrefix(line, "Active Routes:") {
|
||||
if len(lines) <= idx+2 {
|
||||
return nil, "", errNoGateway
|
||||
}
|
||||
|
||||
fields := strings.Fields(lines[idx+2])
|
||||
if len(fields) < 3 {
|
||||
return nil, "", errNoGateway
|
||||
}
|
||||
|
||||
ip := net.ParseIP(fields[2])
|
||||
if ip != nil {
|
||||
return ip, fields[3], nil
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil, "", errNoGateway
|
||||
}
|
||||
|
||||
func parseLinuxIPRoute(output []byte) (net.IP, string, error) {
|
||||
// Linux '/usr/bin/ip route show' format looks like this:
|
||||
// default via 192.168.178.1 dev wlp3s0 metric 303
|
||||
// 192.168.178.0/24 dev wlp3s0 proto kernel scope link src 192.168.178.76 metric 303
|
||||
lines := strings.Split(string(output), "\n")
|
||||
for _, line := range lines {
|
||||
fields := strings.Fields(line)
|
||||
if len(fields) >= 3 && fields[0] == "default" {
|
||||
ip := net.ParseIP(fields[2])
|
||||
if ip != nil {
|
||||
return ip, fields[4], nil
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil, "", errNoGateway
|
||||
}
|
||||
|
||||
func parseLinuxRoute(output []byte) (net.IP, string, error) {
|
||||
// Linux route out format is always like this:
|
||||
// Kernel IP routing table
|
||||
// Destination Gateway Genmask Flags Metric Ref Use Iface
|
||||
// 0.0.0.0 192.168.1.1 0.0.0.0 UG 0 0 0 eth0
|
||||
lines := strings.Split(string(output), "\n")
|
||||
for _, line := range lines {
|
||||
fields := strings.Fields(line)
|
||||
if len(fields) >= 2 && fields[0] == "0.0.0.0" {
|
||||
ip := net.ParseIP(fields[1])
|
||||
if ip != nil {
|
||||
return ip, fields[7], nil
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil, "", errNoGateway
|
||||
}
|
||||
|
||||
func parseDarwinRouteGet(output []byte) (net.IP, string, error) {
|
||||
// Darwin route out format is always like this:
|
||||
// route to: default
|
||||
// destination: default
|
||||
// mask: default
|
||||
// gateway: 192.168.1.1
|
||||
// interface: tun0
|
||||
// flags: <UP,GATEWAY,DONE,STATIC,PRCLONING>
|
||||
lines := strings.Split(string(output), "\n")
|
||||
for _, line := range lines {
|
||||
fields := strings.Fields(line)
|
||||
if len(fields) >= 2 && fields[0] == "gateway:" {
|
||||
ip := net.ParseIP(fields[1])
|
||||
if ip != nil {
|
||||
return ip, "", nil
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil, "", errNoGateway
|
||||
}
|
||||
|
||||
func parseBSDSolarisNetstat(output []byte) (net.IP, string, error) {
|
||||
// netstat -rn produces the following on FreeBSD:
|
||||
// Routing tables
|
||||
//
|
||||
// Internet:
|
||||
// Destination Gateway Flags Netif Expire
|
||||
// default 10.88.88.2 UGS em0
|
||||
// 10.88.88.0/24 link#1 U em0
|
||||
// 10.88.88.148 link#1 UHS lo0
|
||||
// 127.0.0.1 link#2 UH lo0
|
||||
//
|
||||
// Internet6:
|
||||
// Destination Gateway Flags Netif Expire
|
||||
// ::/96 ::1 UGRS lo0
|
||||
// ::1 link#2 UH lo0
|
||||
// ::ffff:0.0.0.0/96 ::1 UGRS lo0
|
||||
// fe80::/10 ::1 UGRS lo0
|
||||
// ...
|
||||
outputLines := strings.Split(string(output), "\n")
|
||||
for _, line := range outputLines {
|
||||
fields := strings.Fields(line)
|
||||
if len(fields) >= 2 && fields[0] == "default" {
|
||||
ip := net.ParseIP(fields[1])
|
||||
if ip != nil {
|
||||
return ip, fields[3], nil
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil, "", errNoGateway
|
||||
}
|
||||
16
net/gateway/gateway_darwin.go
Normal file
16
net/gateway/gateway_darwin.go
Normal file
|
|
@ -0,0 +1,16 @@
|
|||
package gateway
|
||||
|
||||
import (
|
||||
"net"
|
||||
"os/exec"
|
||||
)
|
||||
|
||||
func DiscoverGateway() (net.IP, string, error) {
|
||||
routeCmd := exec.Command("netstat", "-rn")
|
||||
output, err := routeCmd.CombinedOutput()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return parseDarwinRouteGet(output)
|
||||
}
|
||||
16
net/gateway/gateway_freebsd.go
Normal file
16
net/gateway/gateway_freebsd.go
Normal file
|
|
@ -0,0 +1,16 @@
|
|||
package gateway
|
||||
|
||||
import (
|
||||
"net"
|
||||
"os/exec"
|
||||
)
|
||||
|
||||
func DiscoverGateway() (ip net.IP, iface string, err error) {
|
||||
routeCmd := exec.Command("netstat", "-rn")
|
||||
output, err := routeCmd.CombinedOutput()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return parseBSDSolarisNetstat(output)
|
||||
}
|
||||
34
net/gateway/gateway_linux.go
Normal file
34
net/gateway/gateway_linux.go
Normal file
|
|
@ -0,0 +1,34 @@
|
|||
package gateway
|
||||
|
||||
import (
|
||||
"net"
|
||||
"os/exec"
|
||||
)
|
||||
|
||||
func DiscoverGateway() (ip net.IP, iface string, err error) {
|
||||
ip, iface, err = discoverGatewayUsingRoute()
|
||||
if err != nil {
|
||||
ip, iface, err = discoverGatewayUsingIp()
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func discoverGatewayUsingIp() (net.IP, string, error) {
|
||||
routeCmd := exec.Command("ip", "route", "show")
|
||||
output, err := routeCmd.CombinedOutput()
|
||||
if err != nil {
|
||||
return nil, "", err
|
||||
}
|
||||
|
||||
return parseLinuxIPRoute(output)
|
||||
}
|
||||
|
||||
func discoverGatewayUsingRoute() (net.IP, string, error) {
|
||||
routeCmd := exec.Command("route", "-n")
|
||||
output, err := routeCmd.CombinedOutput()
|
||||
if err != nil {
|
||||
return nil, "", err
|
||||
}
|
||||
|
||||
return parseLinuxRoute(output)
|
||||
}
|
||||
16
net/gateway/gateway_solaris.go
Normal file
16
net/gateway/gateway_solaris.go
Normal file
|
|
@ -0,0 +1,16 @@
|
|||
package gateway
|
||||
|
||||
import (
|
||||
"net"
|
||||
"os/exec"
|
||||
)
|
||||
|
||||
func DiscoverGateway() (ip net.IP, iface string, err error) {
|
||||
routeCmd := exec.Command("netstat", "-rn")
|
||||
output, err := routeCmd.CombinedOutput()
|
||||
if err != nil {
|
||||
return nil, "", err
|
||||
}
|
||||
|
||||
return parseBSDSolarisNetstat(output)
|
||||
}
|
||||
248
net/gateway/gateway_test.go
Normal file
248
net/gateway/gateway_test.go
Normal file
|
|
@ -0,0 +1,248 @@
|
|||
package gateway
|
||||
|
||||
import (
|
||||
"net"
|
||||
"testing"
|
||||
)
|
||||
|
||||
type testcase struct {
|
||||
output []byte
|
||||
ok bool
|
||||
gateway string
|
||||
}
|
||||
|
||||
func TestParseWindowsRoutePrint(t *testing.T) {
|
||||
correctData := []byte(`
|
||||
IPv4 Route Table
|
||||
===========================================================================
|
||||
Active Routes:
|
||||
Network Destination Netmask Gateway Interface Metric
|
||||
0.0.0.0 0.0.0.0 10.88.88.2 10.88.88.149 10
|
||||
===========================================================================
|
||||
Persistent Routes:
|
||||
`)
|
||||
randomData := []byte(`
|
||||
Lorem ipsum dolor sit amet, consectetur adipiscing elit,
|
||||
sed do eiusmod tempor incididunt ut labore et dolore magna
|
||||
aliqua. Ut enim ad minim veniam, quis nostrud exercitation
|
||||
`)
|
||||
noRoute := []byte(`
|
||||
IPv4 Route Table
|
||||
===========================================================================
|
||||
Active Routes:
|
||||
`)
|
||||
badRoute1 := []byte(`
|
||||
IPv4 Route Table
|
||||
===========================================================================
|
||||
Active Routes:
|
||||
===========================================================================
|
||||
Persistent Routes:
|
||||
`)
|
||||
badRoute2 := []byte(`
|
||||
IPv4 Route Table
|
||||
===========================================================================
|
||||
Active Routes:
|
||||
Network Destination Netmask Gateway Interface Metric
|
||||
0.0.0.0 0.0.0.0 foo 10.88.88.149 10
|
||||
===========================================================================
|
||||
Persistent Routes:
|
||||
`)
|
||||
|
||||
testcases := []testcase{
|
||||
{correctData, true, "10.88.88.2"},
|
||||
{randomData, false, ""},
|
||||
{noRoute, false, ""},
|
||||
{badRoute1, false, ""},
|
||||
{badRoute2, false, ""},
|
||||
}
|
||||
|
||||
test(t, testcases, parseWindowsRoutePrint)
|
||||
}
|
||||
|
||||
func TestParseLinuxIPRoutePrint(t *testing.T) {
|
||||
correctData := []byte(`
|
||||
default via 192.168.178.1 dev wlp3s0 metric 303
|
||||
192.168.178.0/24 dev wlp3s0 proto kernel scope link src 192.168.178.76 metric 303
|
||||
`)
|
||||
randomData := []byte(`
|
||||
test
|
||||
Lorem ipsum dolor sit amet, consectetur adipiscing elit,
|
||||
sed do eiusmod tempor incididunt ut labore et dolore magna
|
||||
aliqua. Ut enim ad minim veniam, quis nostrud exercitation
|
||||
`)
|
||||
noRoute := []byte(`
|
||||
192.168.178.0/24 dev wlp3s0 proto kernel scope link src 192.168.178.76 metric 303
|
||||
`)
|
||||
badRoute := []byte(`
|
||||
default via foo dev wlp3s0 metric 303
|
||||
192.168.178.0/24 dev wlp3s0 proto kernel scope link src 192.168.178.76 metric 303
|
||||
`)
|
||||
|
||||
testcases := []testcase{
|
||||
{correctData, true, "192.168.178.1"},
|
||||
{randomData, false, ""},
|
||||
{noRoute, false, ""},
|
||||
{badRoute, false, ""},
|
||||
}
|
||||
|
||||
test(t, testcases, parseLinuxIPRoute)
|
||||
}
|
||||
|
||||
func TestParseLinuxRoutePrint(t *testing.T) {
|
||||
correctData := []byte(`
|
||||
Kernel IP routing table
|
||||
Destination Gateway Genmask Flags Metric Ref Use Iface
|
||||
0.0.0.0 192.168.1.1 0.0.0.0 UG 0 0 0 eth0
|
||||
`)
|
||||
randomData := []byte(`
|
||||
test
|
||||
Lorem ipsum dolor sit amet, consectetur adipiscing elit,
|
||||
sed do eiusmod tempor incididunt ut labore et dolore magna
|
||||
aliqua. Ut enim ad minim veniam, quis nostrud exercitation
|
||||
`)
|
||||
noRoute := []byte(`
|
||||
Kernel IP routing table
|
||||
Destination Gateway Genmask Flags Metric Ref Use Iface
|
||||
`)
|
||||
badRoute := []byte(`
|
||||
Kernel IP routing table
|
||||
Destination Gateway Genmask Flags Metric Ref Use Iface
|
||||
0.0.0.0 foo 0.0.0.0 UG 0 0 0 eth0
|
||||
`)
|
||||
|
||||
testcases := []testcase{
|
||||
{correctData, true, "192.168.1.1"},
|
||||
{randomData, false, ""},
|
||||
{noRoute, false, ""},
|
||||
{badRoute, false, ""},
|
||||
}
|
||||
|
||||
test(t, testcases, parseLinuxRoute)
|
||||
}
|
||||
|
||||
func TestParseDarwinRouteGet(t *testing.T) {
|
||||
correctData := []byte(`
|
||||
route to: 0.0.0.0
|
||||
destination: default
|
||||
mask: default
|
||||
gateway: 172.16.32.1
|
||||
interface: en0
|
||||
flags: <UP,GATEWAY,DONE,STATIC,PRCLONING>
|
||||
recvpipe sendpipe ssthresh rtt,msec rttvar hopcount mtu expire
|
||||
0 0 0 0 0 0 1500 0
|
||||
`)
|
||||
randomData := []byte(`
|
||||
test
|
||||
Lorem ipsum dolor sit amet, consectetur adipiscing elit,
|
||||
sed do eiusmod tempor incididunt ut labore et dolore magna
|
||||
aliqua. Ut enim ad minim veniam, quis nostrud exercitation
|
||||
`)
|
||||
noRoute := []byte(`
|
||||
route to: 0.0.0.0
|
||||
destination: default
|
||||
mask: default
|
||||
`)
|
||||
badRoute := []byte(`
|
||||
route to: 0.0.0.0
|
||||
destination: default
|
||||
mask: default
|
||||
gateway: foo
|
||||
interface: en0
|
||||
flags: <UP,GATEWAY,DONE,STATIC,PRCLONING>
|
||||
recvpipe sendpipe ssthresh rtt,msec rttvar hopcount mtu expire
|
||||
0 0 0 0 0 0 1500 0
|
||||
`)
|
||||
|
||||
testcases := []testcase{
|
||||
{correctData, true, "172.16.32.1"},
|
||||
{randomData, false, ""},
|
||||
{noRoute, false, ""},
|
||||
{badRoute, false, ""},
|
||||
}
|
||||
|
||||
test(t, testcases, parseDarwinRouteGet)
|
||||
}
|
||||
|
||||
func TestParseBSDSolarisNetstat(t *testing.T) {
|
||||
correctDataFreeBSD := []byte(`
|
||||
Routing tables
|
||||
|
||||
Internet:
|
||||
Destination Gateway Flags Netif Expire
|
||||
default 10.88.88.2 UGS em0
|
||||
10.88.88.0/24 link#1 U em0
|
||||
10.88.88.148 link#1 UHS lo0
|
||||
127.0.0.1 link#2 UH lo0
|
||||
|
||||
Internet6:
|
||||
Destination Gateway Flags Netif Expire
|
||||
::/96 ::1 UGRS lo0
|
||||
::1 link#2 UH lo0
|
||||
::ffff:0.0.0.0/96 ::1 UGRS lo0
|
||||
fe80::/10 ::1 UGRS lo0
|
||||
`)
|
||||
correctDataSolaris := []byte(`
|
||||
Routing Table: IPv4
|
||||
Destination Gateway Flags Ref Use Interface
|
||||
-------------------- -------------------- ----- ----- ---------- ---------
|
||||
default 172.16.32.1 UG 2 76419 net0
|
||||
127.0.0.1 127.0.0.1 UH 2 36 lo0
|
||||
172.16.32.0 172.16.32.17 U 4 8100 net0
|
||||
|
||||
Routing Table: IPv6
|
||||
Destination/Mask Gateway Flags Ref Use If
|
||||
--------------------------- --------------------------- ----- --- ------- -----
|
||||
::1 ::1 UH 3 75382 lo0
|
||||
2001:470:deeb:32::/64 2001:470:deeb:32::17 U 3 2744 net0
|
||||
fe80::/10 fe80::6082:52ff:fedc:7df0 U 3 8430 net0
|
||||
`)
|
||||
randomData := []byte(`
|
||||
Lorem ipsum dolor sit amet, consectetur adipiscing elit,
|
||||
sed do eiusmod tempor incididunt ut labore et dolore magna
|
||||
aliqua. Ut enim ad minim veniam, quis nostrud exercitation
|
||||
`)
|
||||
noRoute := []byte(`
|
||||
Internet:
|
||||
Destination Gateway Flags Netif Expire
|
||||
10.88.88.0/24 link#1 U em0
|
||||
10.88.88.148 link#1 UHS lo0
|
||||
127.0.0.1 link#2 UH lo0
|
||||
`)
|
||||
badRoute := []byte(`
|
||||
Internet:
|
||||
Destination Gateway Flags Netif Expire
|
||||
default foo UGS em0
|
||||
10.88.88.0/24 link#1 U em0
|
||||
10.88.88.148 link#1 UHS lo0
|
||||
127.0.0.1 link#2 UH lo0
|
||||
`)
|
||||
|
||||
testcases := []testcase{
|
||||
{correctDataFreeBSD, true, "10.88.88.2"},
|
||||
{correctDataSolaris, true, "172.16.32.1"},
|
||||
{randomData, false, ""},
|
||||
{noRoute, false, ""},
|
||||
{badRoute, false, ""},
|
||||
}
|
||||
|
||||
test(t, testcases, parseBSDSolarisNetstat)
|
||||
}
|
||||
|
||||
func test(t *testing.T, testcases []testcase, fn func([]byte) (net.IP, string, error)) {
|
||||
for i, tc := range testcases {
|
||||
net, iface, err := fn(tc.output)
|
||||
if tc.ok {
|
||||
if err != nil {
|
||||
t.Errorf("Unexpected error in test #%d: %v", i, err)
|
||||
}
|
||||
if net.String() != tc.gateway {
|
||||
t.Errorf("Unexpected gateway address %v != %s", net, tc.gateway)
|
||||
}
|
||||
if "" == iface {
|
||||
t.Errorf("Unexpected interface")
|
||||
}
|
||||
} else if err == nil {
|
||||
t.Errorf("Unexpected nil error in test #%d", i)
|
||||
}
|
||||
}
|
||||
}
|
||||
14
net/gateway/gateway_unimplemented.go
Normal file
14
net/gateway/gateway_unimplemented.go
Normal file
|
|
@ -0,0 +1,14 @@
|
|||
// +build !darwin,!linux,!windows,!solaris,!freebsd
|
||||
|
||||
package gateway
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
"runtime"
|
||||
)
|
||||
|
||||
func DiscoverGateway() (ip net.IP, iface string, err error) {
|
||||
err = fmt.Errorf("DiscoverGateway not implemented for OS %s", runtime.GOOS)
|
||||
return
|
||||
}
|
||||
16
net/gateway/gateway_windows.go
Normal file
16
net/gateway/gateway_windows.go
Normal file
|
|
@ -0,0 +1,16 @@
|
|||
package gateway
|
||||
|
||||
import (
|
||||
"net"
|
||||
"os/exec"
|
||||
)
|
||||
|
||||
func DiscoverGateway() (ip net.IP, iface string, err error) {
|
||||
routeCmd := exec.Command("route", "print", "0.0.0.0")
|
||||
output, err := routeCmd.CombinedOutput()
|
||||
if err != nil {
|
||||
return nil, "", err
|
||||
}
|
||||
|
||||
return parseWindowsRoutePrint(output)
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user