mirror of
https://github.com/net-byte/vtun
synced 2024-03-14 10:50:03 +08:00
74 lines
2.6 KiB
Go
74 lines
2.6 KiB
Go
package tun
|
|
|
|
import (
|
|
"log"
|
|
"net"
|
|
"runtime"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/net-byte/vtun/common/config"
|
|
"github.com/net-byte/vtun/common/netutil"
|
|
"github.com/songgao/water"
|
|
)
|
|
|
|
func CreateTun(config config.Config) (iface *water.Interface) {
|
|
c := water.Config{DeviceType: water.TUN}
|
|
iface, err := water.New(c)
|
|
if err != nil {
|
|
log.Fatalln("failed to create tun interface:", err)
|
|
}
|
|
log.Println("interface created:", iface.Name())
|
|
configTun(config, iface)
|
|
return iface
|
|
}
|
|
|
|
func configTun(config config.Config, iface *water.Interface) {
|
|
os := runtime.GOOS
|
|
ip, _, err := net.ParseCIDR(config.CIDR)
|
|
if err != nil {
|
|
log.Panicf("error cidr %v", config.CIDR)
|
|
}
|
|
if os == "linux" {
|
|
netutil.ExecCmd("/sbin/ip", "link", "set", "dev", iface.Name(), "mtu", strconv.Itoa(config.MTU))
|
|
netutil.ExecCmd("/sbin/ip", "addr", "add", config.CIDR, "dev", iface.Name())
|
|
netutil.ExecCmd("/sbin/ip", "link", "set", "dev", iface.Name(), "up")
|
|
if !config.ServerMode && config.GlobalMode {
|
|
physicalIface := netutil.GetPhysicalInterface()
|
|
serverIP := netutil.LookupIP(strings.Split(config.ServerAddr, ":")[0])
|
|
if physicalIface != "" && serverIP != "" {
|
|
netutil.ExecCmd("/sbin/ip", "route", "add", "0.0.0.0/1", "dev", iface.Name())
|
|
netutil.ExecCmd("/sbin/ip", "route", "add", "128.0.0.0/1", "dev", iface.Name())
|
|
netutil.ExecCmd("/sbin/ip", "route", "add", "8.8.8.8/32", "via", config.DefaultGateway, "dev", physicalIface)
|
|
netutil.ExecCmd("/sbin/ip", "route", "add", strings.Join([]string{serverIP, "32"}, "/"), "via", config.DefaultGateway, "dev", physicalIface)
|
|
}
|
|
}
|
|
|
|
} else if os == "darwin" {
|
|
gateway := config.IntranetServerIP
|
|
netutil.ExecCmd("ifconfig", iface.Name(), "inet", ip.String(), gateway, "up")
|
|
if !config.ServerMode && config.GlobalMode {
|
|
physicalIface := netutil.GetPhysicalInterface()
|
|
serverIP := netutil.LookupIP(strings.Split(config.ServerAddr, ":")[0])
|
|
if physicalIface != "" && serverIP != "" {
|
|
netutil.ExecCmd("route", "add", serverIP, config.DefaultGateway)
|
|
netutil.ExecCmd("route", "add", "8.8.8.8", config.DefaultGateway)
|
|
netutil.ExecCmd("route", "add", "0.0.0.0/1", "-interface", iface.Name())
|
|
netutil.ExecCmd("route", "add", "128.0.0.0/1", "-interface", iface.Name())
|
|
netutil.ExecCmd("route", "add", "default", gateway)
|
|
netutil.ExecCmd("route", "change", "default", gateway)
|
|
}
|
|
}
|
|
} else {
|
|
log.Printf("not support os:%v", os)
|
|
}
|
|
}
|
|
|
|
func Reset(config config.Config) {
|
|
os := runtime.GOOS
|
|
if os == "darwin" && !config.ServerMode && config.GlobalMode {
|
|
netutil.ExecCmd("route", "add", "default", config.DefaultGateway)
|
|
netutil.ExecCmd("route", "change", "default", config.DefaultGateway)
|
|
}
|
|
}
|