134 lines
2.6 KiB
Go
134 lines
2.6 KiB
Go
//go:build !linux && !windows && !darwin
|
|
// +build !linux,!windows,!darwin
|
|
|
|
package gost
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"github.com/go-log/log"
|
|
"github.com/songgao/water"
|
|
)
|
|
|
|
func createTun(cfg TunConfig) (conn net.Conn, itf *net.Interface, err error) {
|
|
ip, _, err := net.ParseCIDR(cfg.Addr)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
ifce, err := water.New(water.Config{
|
|
DeviceType: water.TUN,
|
|
})
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
mtu := cfg.MTU
|
|
if mtu <= 0 {
|
|
mtu = DefaultMTU
|
|
}
|
|
|
|
cmd := fmt.Sprintf("ifconfig %s inet %s mtu %d up", ifce.Name(), cfg.Addr, mtu)
|
|
log.Log("[tun]", cmd)
|
|
args := strings.Split(cmd, " ")
|
|
if er := exec.Command(args[0], args[1:]...).Run(); er != nil {
|
|
err = fmt.Errorf("%s: %v", cmd, er)
|
|
return
|
|
}
|
|
|
|
if err = addTunRoutes(ifce.Name(), cfg.Routes...); err != nil {
|
|
return
|
|
}
|
|
|
|
itf, err = net.InterfaceByName(ifce.Name())
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
conn = &tunTapConn{
|
|
ifce: ifce,
|
|
addr: &net.IPAddr{IP: ip},
|
|
}
|
|
return
|
|
}
|
|
|
|
func createTap(cfg TapConfig) (conn net.Conn, itf *net.Interface, err error) {
|
|
ip, _, _ := net.ParseCIDR(cfg.Addr)
|
|
|
|
ifce, err := water.New(water.Config{
|
|
DeviceType: water.TAP,
|
|
})
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
mtu := cfg.MTU
|
|
if mtu <= 0 {
|
|
mtu = DefaultMTU
|
|
}
|
|
|
|
var cmd string
|
|
if cfg.Addr != "" {
|
|
cmd = fmt.Sprintf("ifconfig %s inet %s mtu %d up", ifce.Name(), cfg.Addr, mtu)
|
|
} else {
|
|
cmd = fmt.Sprintf("ifconfig %s mtu %d up", ifce.Name(), mtu)
|
|
}
|
|
log.Log("[tap]", cmd)
|
|
args := strings.Split(cmd, " ")
|
|
if er := exec.Command(args[0], args[1:]...).Run(); er != nil {
|
|
err = fmt.Errorf("%s: %v", cmd, er)
|
|
return
|
|
}
|
|
|
|
if err = addTapRoutes(ifce.Name(), cfg.Gateway, cfg.Routes...); err != nil {
|
|
return
|
|
}
|
|
|
|
itf, err = net.InterfaceByName(ifce.Name())
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
conn = &tunTapConn{
|
|
ifce: ifce,
|
|
addr: &net.IPAddr{IP: ip},
|
|
}
|
|
return
|
|
}
|
|
|
|
func addTunRoutes(ifName string, routes ...IPRoute) error {
|
|
for _, route := range routes {
|
|
if route.Dest == nil {
|
|
continue
|
|
}
|
|
cmd := fmt.Sprintf("route add -net %s -interface %s", route.Dest.String(), ifName)
|
|
log.Logf("[tun] %s", cmd)
|
|
args := strings.Split(cmd, " ")
|
|
if er := exec.Command(args[0], args[1:]...).Run(); er != nil {
|
|
return fmt.Errorf("%s: %v", cmd, er)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func addTapRoutes(ifName string, gw string, routes ...string) error {
|
|
for _, route := range routes {
|
|
if route == "" {
|
|
continue
|
|
}
|
|
cmd := fmt.Sprintf("route add -net %s dev %s", route, ifName)
|
|
if gw != "" {
|
|
cmd += " gw " + gw
|
|
}
|
|
log.Logf("[tap] %s", cmd)
|
|
args := strings.Split(cmd, " ")
|
|
if er := exec.Command(args[0], args[1:]...).Run(); er != nil {
|
|
return fmt.Errorf("%s: %v", cmd, er)
|
|
}
|
|
}
|
|
return nil
|
|
}
|