116 lines
2.4 KiB
Go
116 lines
2.4 KiB
Go
package tun
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
|
|
"github.com/go-gost/core/logger"
|
|
mdata "github.com/go-gost/core/metadata"
|
|
mdutil "github.com/go-gost/core/metadata/util"
|
|
"github.com/go-gost/core/router"
|
|
tun_util "github.com/go-gost/x/internal/util/tun"
|
|
"github.com/go-gost/x/registry"
|
|
xrouter "github.com/go-gost/x/router"
|
|
)
|
|
|
|
const (
|
|
defaultMTU = 1350
|
|
defaultReadBufferSize = 4096
|
|
)
|
|
|
|
type metadata struct {
|
|
config *tun_util.Config
|
|
readBufferSize int
|
|
}
|
|
|
|
func (l *tunListener) parseMetadata(md mdata.Metadata) (err error) {
|
|
const (
|
|
name = "name"
|
|
netKey = "net"
|
|
peer = "peer"
|
|
mtu = "mtu"
|
|
route = "route"
|
|
routes = "routes"
|
|
gateway = "gw"
|
|
)
|
|
|
|
l.md.readBufferSize = mdutil.GetInt(md, "tun.rbuf", "rbuf", "readBufferSize")
|
|
if l.md.readBufferSize <= 0 {
|
|
l.md.readBufferSize = defaultReadBufferSize
|
|
}
|
|
|
|
config := &tun_util.Config{
|
|
Name: mdutil.GetString(md, name),
|
|
Peer: mdutil.GetString(md, peer),
|
|
MTU: mdutil.GetInt(md, mtu),
|
|
Router: registry.RouterRegistry().Get(mdutil.GetString(md, "router")),
|
|
}
|
|
if config.MTU <= 0 {
|
|
config.MTU = defaultMTU
|
|
}
|
|
if gw := mdutil.GetString(md, gateway); gw != "" {
|
|
config.Gateway = net.ParseIP(gw)
|
|
}
|
|
|
|
for _, s := range strings.Split(mdutil.GetString(md, netKey), ",") {
|
|
if s = strings.TrimSpace(s); s == "" {
|
|
continue
|
|
}
|
|
ip, ipNet, err := net.ParseCIDR(s)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
config.Net = append(config.Net, net.IPNet{
|
|
IP: ip,
|
|
Mask: ipNet.Mask,
|
|
})
|
|
}
|
|
|
|
for _, s := range strings.Split(mdutil.GetString(md, route), ",") {
|
|
_, ipNet, _ := net.ParseCIDR(strings.TrimSpace(s))
|
|
if ipNet == nil {
|
|
continue
|
|
}
|
|
|
|
l.routes = append(l.routes, &router.Route{
|
|
Net: ipNet,
|
|
Gateway: config.Gateway,
|
|
})
|
|
}
|
|
|
|
for _, s := range mdutil.GetStrings(md, routes) {
|
|
ss := strings.SplitN(s, " ", 2)
|
|
if len(ss) == 2 {
|
|
var route router.Route
|
|
_, ipNet, _ := net.ParseCIDR(strings.TrimSpace(ss[0]))
|
|
if ipNet == nil {
|
|
continue
|
|
}
|
|
route.Net = ipNet
|
|
gw := net.ParseIP(ss[1])
|
|
if gw == nil {
|
|
gw = config.Gateway
|
|
}
|
|
|
|
l.routes = append(l.routes, &router.Route{
|
|
Net: ipNet,
|
|
Gateway: gw,
|
|
})
|
|
}
|
|
}
|
|
|
|
if config.Router == nil && len(l.routes) > 0 {
|
|
config.Router = xrouter.NewRouter(
|
|
xrouter.RoutesOption(l.routes),
|
|
xrouter.LoggerOption(logger.Default().WithFields(map[string]any{
|
|
"kind": "router",
|
|
"router": "@internal",
|
|
})),
|
|
)
|
|
}
|
|
|
|
l.md.config = config
|
|
|
|
return
|
|
}
|