apps/pg
1
0
Fork 0
mirror of https://github.com/sigcn/pg.git synced 2025-10-27 22:31:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

vpn/iface: merge ipv4/ipv6 routing

Browse Source
This commit is contained in:
rkonfj
2024-06-01 13:54:19 +08:00
parent af9b20857f
commit 9a511afaec
2 changed files with 15 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
vpn/iface/iface.go
View File

@@ -28,8 +28,7 @@ var _ RoutingTable = (*TunInterface)(nil)
type TunInterface struct {
dev tun.Device
ifName string
ipv6 *lru.Cache[string, []*net.IPNet]
ipv4 *lru.Cache[string, []*net.IPNet]
routing *lru.Cache[string, []*net.IPNet]
peers *lru.Cache[string, net.Addr]
peersMutex sync.RWMutex
}
@@ -48,8 +47,7 @@ func Create(tunName string, cfg Config) (*TunInterface, error) {
return &TunInterface{
dev: device,
ifName: tunName,
ipv6: lru.New[string, []*net.IPNet](256),
ipv4: lru.New[string, []*net.IPNet](256),
routing: lru.New[string, []*net.IPNet](512),
peers: lru.New[string, net.Addr](1024),
}, nil
}
@@ -62,18 +60,7 @@ func (r *TunInterface) GetPeer(ip string) (net.Addr, bool) {
return peerID, true
}
dstIP := net.ParseIP(ip)
if dstIP.To4() != nil {
k, _, _ := r.ipv4.Find(func(k string, v []*net.IPNet) bool {
for _, cidr := range v {
if cidr.Contains(dstIP) {
return true
}
}
return false
})
return r.peers.Get(k)
}
k, _, _ := r.ipv6.Find(func(k string, v []*net.IPNet) bool {
k, _, _ := r.routing.Find(func(k string, v []*net.IPNet) bool {
for _, cidr := range v {
if cidr.Contains(dstIP) {
return true
@@ -98,26 +85,14 @@ func (r *TunInterface) AddPeer(ipv4, ipv6 string, peer net.Addr) {
func (r *TunInterface) AddRoute(cidr *net.IPNet, via net.IP) {
r.peersMutex.Lock()
defer r.peersMutex.Unlock()
var cidrs []*net.IPNet
if via.To4() != nil {
cidrs, _ = r.ipv4.Get(via.String())
cidrs, _ := r.routing.Get(via.String())
for _, cmp := range cidrs {
if cmp.String() == cidr.String() {
return
}
}
cidrs = append(cidrs, cidr)
r.ipv4.Put(via.String(), cidrs)
} else {
cidrs, _ := r.ipv6.Get(via.String())
for _, cmp := range cidrs {
if cmp.String() == cidr.String() {
return
}
}
cidrs = append(cidrs, cidr)
r.ipv6.Put(via.String(), cidrs)
}
r.routing.Put(via.String(), cidrs)
r.updateRoute(via, cidrs[:len(cidrs)-1], cidrs)
}

3
vpn/iface/iface_unix.go
View File

@@ -23,8 +23,7 @@ func CreateFD(tunFD int, cfg Config) (*TunInterface, error) {
}
return &TunInterface{
dev: device,
ipv6: lru.New[string, []*net.IPNet](256),
ipv4: lru.New[string, []*net.IPNet](256),
routing: lru.New[string, []*net.IPNet](512),
peers: lru.New[string, net.Addr](1024),
}, nil
}