mirror of
https://github.com/kubenetworks/kubevpn.git
synced 2025-12-24 11:51:13 +08:00
optimize structure
This commit is contained in:
@@ -1,62 +0,0 @@
|
||||
package pkg
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"github.com/wencaiwulue/kubevpn/tun"
|
||||
"net"
|
||||
"os"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func parseIP(s string, port string) (ips []string) {
|
||||
if s == "" {
|
||||
return
|
||||
}
|
||||
if port == "" {
|
||||
port = "8080" // default port
|
||||
}
|
||||
|
||||
file, err := os.Open(s)
|
||||
if err != nil {
|
||||
ss := strings.Split(s, ",")
|
||||
for _, s := range ss {
|
||||
s = strings.TrimSpace(s)
|
||||
if s != "" {
|
||||
// TODO: support IPv6
|
||||
if !strings.Contains(s, ":") {
|
||||
s = s + ":" + port
|
||||
}
|
||||
ips = append(ips, s)
|
||||
}
|
||||
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
scanner := bufio.NewScanner(file)
|
||||
for scanner.Scan() {
|
||||
line := strings.TrimSpace(scanner.Text())
|
||||
if line == "" || strings.HasPrefix(line, "#") {
|
||||
continue
|
||||
}
|
||||
if !strings.Contains(line, ":") {
|
||||
line = line + ":" + port
|
||||
}
|
||||
ips = append(ips, line)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func parseIPRoutes(routeStringList string) (routes []tun.IPRoute) {
|
||||
if len(routeStringList) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
ss := strings.Split(routeStringList, ",")
|
||||
for _, s := range ss {
|
||||
if _, inet, _ := net.ParseCIDR(strings.TrimSpace(s)); inet != nil {
|
||||
routes = append(routes, tun.IPRoute{Dest: inet})
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
15
pkg/route.go
15
pkg/route.go
@@ -7,6 +7,7 @@ import (
|
||||
"github.com/wencaiwulue/kubevpn/core"
|
||||
"github.com/wencaiwulue/kubevpn/tun"
|
||||
"net"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type route struct {
|
||||
@@ -130,3 +131,17 @@ func (r *router) Close() error {
|
||||
}
|
||||
return r.server.Close()
|
||||
}
|
||||
|
||||
func parseIPRoutes(routeStringList string) (routes []tun.IPRoute) {
|
||||
if len(routeStringList) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
ss := strings.Split(routeStringList, ",")
|
||||
for _, s := range ss {
|
||||
if _, inet, _ := net.ParseCIDR(strings.TrimSpace(s)); inet != nil {
|
||||
routes = append(routes, tun.IPRoute{Dest: inet})
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user