mirror of
https://github.com/luscis/openlan.git
synced 2025-11-02 21:34:00 +08:00
clone from danieldin95
This commit is contained in:
sicheng
310
pkg/access/point_linux.go
Executable file
310
pkg/access/point_linux.go
Executable file
@@ -0,0 +1,310 @@
|
||||
package access
|
||||
|
||||
import (
|
||||
"github.com/luscis/openlan/pkg/config"
|
||||
"github.com/luscis/openlan/pkg/models"
|
||||
"github.com/luscis/openlan/pkg/network"
|
||||
"github.com/vishvananda/netlink"
|
||||
"net"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Point struct {
|
||||
MixPoint
|
||||
// private
|
||||
brName string
|
||||
ipMtu int
|
||||
addr string
|
||||
bypass *netlink.Route
|
||||
routes []*models.Route
|
||||
link netlink.Link
|
||||
uuid string
|
||||
}
|
||||
|
||||
func NewPoint(config *config.Point) *Point {
|
||||
ipMtu := config.Interface.IPMtu
|
||||
if ipMtu == 0 {
|
||||
ipMtu = 1500
|
||||
}
|
||||
p := Point{
|
||||
ipMtu: ipMtu,
|
||||
brName: config.Interface.Bridge,
|
||||
MixPoint: NewMixPoint(config),
|
||||
}
|
||||
return &p
|
||||
}
|
||||
|
||||
func (p *Point) Initialize() {
|
||||
p.worker.listener.AddAddr = p.AddAddr
|
||||
p.worker.listener.DelAddr = p.DelAddr
|
||||
p.worker.listener.AddRoutes = p.AddRoutes
|
||||
p.worker.listener.DelRoutes = p.DelRoutes
|
||||
p.worker.listener.OnTap = p.OnTap
|
||||
p.MixPoint.Initialize()
|
||||
}
|
||||
|
||||
func (p *Point) DelAddr(ipStr string) error {
|
||||
if p.link == nil || ipStr == "" {
|
||||
return nil
|
||||
}
|
||||
ipAddr, err := netlink.ParseAddr(ipStr)
|
||||
if err != nil {
|
||||
p.out.Error("Point.AddAddr.ParseCIDR %s: %s", ipStr, err)
|
||||
return err
|
||||
}
|
||||
if err := netlink.AddrDel(p.link, ipAddr); err != nil {
|
||||
p.out.Warn("Point.DelAddr.UnsetLinkIp: %s", err)
|
||||
}
|
||||
p.out.Info("Point.DelAddr: %s", ipStr)
|
||||
p.addr = ""
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *Point) AddAddr(ipStr string) error {
|
||||
if ipStr == "" || p.link == nil {
|
||||
return nil
|
||||
}
|
||||
ipAddr, err := netlink.ParseAddr(ipStr)
|
||||
if err != nil {
|
||||
p.out.Error("Point.AddAddr.ParseCIDR %s: %s", ipStr, err)
|
||||
return err
|
||||
}
|
||||
if err := netlink.AddrAdd(p.link, ipAddr); err != nil {
|
||||
p.out.Warn("Point.AddAddr.SetLinkIp: %s", err)
|
||||
return err
|
||||
}
|
||||
p.out.Info("Point.AddAddr: %s", ipStr)
|
||||
p.addr = ipStr
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *Point) UpBr(name string) *netlink.Bridge {
|
||||
if name == "" {
|
||||
return nil
|
||||
}
|
||||
la := netlink.LinkAttrs{TxQLen: -1, Name: name}
|
||||
br := &netlink.Bridge{LinkAttrs: la}
|
||||
if link, err := netlink.LinkByName(name); link == nil {
|
||||
p.out.Warn("Point.UpBr: %s %s", name, err)
|
||||
err := netlink.LinkAdd(br)
|
||||
if err != nil {
|
||||
p.out.Warn("Point.UpBr.newBr: %s %s", name, err)
|
||||
}
|
||||
}
|
||||
link, err := netlink.LinkByName(name)
|
||||
if link == nil {
|
||||
p.out.Error("Point.UpBr: %s %s", name, err)
|
||||
return nil
|
||||
}
|
||||
if err := netlink.LinkSetUp(link); err != nil {
|
||||
p.out.Error("Point.UpBr.LinkUp: %s", err)
|
||||
}
|
||||
return br
|
||||
}
|
||||
|
||||
func (p *Point) OnTap(w *TapWorker) error {
|
||||
p.out.Info("Point.OnTap")
|
||||
tap := w.device
|
||||
name := tap.Name()
|
||||
if tap.Type() == network.ProviderVir { // virtual device
|
||||
p.out.Error("Point.OnTap: not support %s", tap.Type())
|
||||
return nil
|
||||
}
|
||||
// kernel device
|
||||
link, err := netlink.LinkByName(name)
|
||||
if err != nil {
|
||||
p.out.Error("Point.OnTap: Get %s: %s", name, err)
|
||||
return err
|
||||
}
|
||||
if err := netlink.LinkSetMTU(link, p.ipMtu); err != nil {
|
||||
p.out.Error("Point.OnTap.SetMTU: %s", err)
|
||||
}
|
||||
if br := p.UpBr(p.brName); br != nil {
|
||||
if err := netlink.LinkSetMaster(link, br); err != nil {
|
||||
p.out.Error("Point.OnTap.AddSlave: Switch dev %s: %s", name, err)
|
||||
}
|
||||
link, err = netlink.LinkByName(p.brName)
|
||||
if err != nil {
|
||||
p.out.Error("Point.OnTap: Get %s: %s", p.brName, err)
|
||||
}
|
||||
}
|
||||
if p.config.Interface.Cost > 0 {
|
||||
port := network.NewBrPort(name)
|
||||
if err := port.Cost(p.config.Interface.Cost); err != nil {
|
||||
p.out.Error("Point.OnTap: Cost %s: %s", err)
|
||||
}
|
||||
}
|
||||
p.link = link
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *Point) GetRemote() string {
|
||||
conn := p.worker.conWorker
|
||||
if conn == nil {
|
||||
return ""
|
||||
}
|
||||
remote := conn.client.RemoteAddr()
|
||||
remote = strings.SplitN(remote, ":", 2)[0]
|
||||
return remote
|
||||
}
|
||||
|
||||
func (p *Point) AddBypass(routes []*models.Route) {
|
||||
remote := p.GetRemote()
|
||||
if !p.config.ByPass {
|
||||
return
|
||||
}
|
||||
addr, dest, _ := net.ParseCIDR(remote + "/32")
|
||||
gws, err := netlink.RouteGet(addr)
|
||||
if err != nil || len(gws) == 0 {
|
||||
p.out.Error("Point.AddBypass: RouteGet %s: %s", addr, err)
|
||||
return
|
||||
}
|
||||
rt := &netlink.Route{
|
||||
LinkIndex: gws[0].LinkIndex,
|
||||
Dst: dest,
|
||||
Gw: gws[0].Gw,
|
||||
Table: 100,
|
||||
}
|
||||
p.out.Debug("Point.AddBypass: %s")
|
||||
if err := netlink.RouteReplace(rt); err != nil {
|
||||
p.out.Warn("Point.AddBypass: %s %s", rt.Dst, err)
|
||||
return
|
||||
}
|
||||
p.out.Info("Point.AddBypass: route %s via %s", rt.Dst, rt.Gw)
|
||||
ru := netlink.NewRule()
|
||||
ru.Table = 100
|
||||
ru.Priority = 16383
|
||||
if err := netlink.RuleAdd(ru); err != nil {
|
||||
p.out.Warn("Point.AddBypass: %s %s", ru.Dst, err)
|
||||
}
|
||||
p.out.Info("Point.AddBypass: %s", ru)
|
||||
p.bypass = rt
|
||||
for _, rt := range routes {
|
||||
if rt.Prefix != "0.0.0.0/0" {
|
||||
continue
|
||||
}
|
||||
gw := net.ParseIP(rt.NextHop)
|
||||
_, dst0, _ := net.ParseCIDR("0.0.0.0/1")
|
||||
rt0 := netlink.Route{
|
||||
LinkIndex: p.link.Attrs().Index,
|
||||
Dst: dst0,
|
||||
Gw: gw,
|
||||
Priority: rt.Metric,
|
||||
}
|
||||
p.out.Debug("Point.AddBypass: %s", rt0)
|
||||
if err := netlink.RouteAdd(&rt0); err != nil {
|
||||
p.out.Warn("Point.AddBypass: %s %s", rt0.Dst, err)
|
||||
}
|
||||
p.out.Info("Point.AddBypass: route %s via %s", rt0.Dst, rt0.Gw)
|
||||
_, dst1, _ := net.ParseCIDR("128.0.0.0/1")
|
||||
rt1 := netlink.Route{
|
||||
LinkIndex: p.link.Attrs().Index,
|
||||
Dst: dst1,
|
||||
Gw: gw,
|
||||
Priority: rt.Metric,
|
||||
}
|
||||
p.out.Debug("Point.AddBypass: %s", rt1)
|
||||
if err := netlink.RouteAdd(&rt1); err != nil {
|
||||
p.out.Warn("Point.AddBypass: %s %s", rt1.Dst, err)
|
||||
}
|
||||
p.out.Info("Point.AddBypass: route %s via %s", rt1.Dst, rt1.Gw)
|
||||
}
|
||||
}
|
||||
|
||||
func (p *Point) AddRoutes(routes []*models.Route) error {
|
||||
if routes == nil || p.link == nil {
|
||||
return nil
|
||||
}
|
||||
p.AddBypass(routes)
|
||||
for _, rt := range routes {
|
||||
_, dst, err := net.ParseCIDR(rt.Prefix)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
nxt := net.ParseIP(rt.NextHop)
|
||||
rte := netlink.Route{
|
||||
LinkIndex: p.link.Attrs().Index,
|
||||
Dst: dst,
|
||||
Gw: nxt,
|
||||
Priority: rt.Metric,
|
||||
}
|
||||
p.out.Debug("Point.AddRoute: %s", rte)
|
||||
if err := netlink.RouteAdd(&rte); err != nil {
|
||||
p.out.Warn("Point.AddRoute: %s %s", rt.Prefix, err)
|
||||
continue
|
||||
}
|
||||
p.out.Info("Point.AddRoutes: route %s via %s", rt.Prefix, rt.NextHop)
|
||||
}
|
||||
p.routes = routes
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *Point) DelBypass(routes []*models.Route) {
|
||||
if !p.config.ByPass || p.bypass == nil {
|
||||
return
|
||||
}
|
||||
p.out.Debug("Point.DelRoute: %s")
|
||||
rt := p.bypass
|
||||
if err := netlink.RouteAdd(rt); err != nil {
|
||||
p.out.Warn("Point.DelRoute: %s %s", rt.Dst, err)
|
||||
}
|
||||
p.out.Info("Point.DelBypass: route %s via %s", rt.Dst, rt.Gw)
|
||||
p.bypass = nil
|
||||
for _, rt := range routes {
|
||||
if rt.Prefix != "0.0.0.0/0" {
|
||||
continue
|
||||
}
|
||||
gw := net.ParseIP(rt.NextHop)
|
||||
_, dst0, _ := net.ParseCIDR("0.0.0.0/1")
|
||||
rt0 := netlink.Route{
|
||||
LinkIndex: p.link.Attrs().Index,
|
||||
Dst: dst0,
|
||||
Gw: gw,
|
||||
Priority: rt.Metric,
|
||||
}
|
||||
p.out.Debug("Point.DelBypass: %s", rt0)
|
||||
if err := netlink.RouteDel(&rt0); err != nil {
|
||||
p.out.Warn("Point.DelBypass: %s %s", rt0.Dst, err)
|
||||
}
|
||||
p.out.Info("Point.DelBypass: route %s via %s", rt0.Dst, rt0.Gw)
|
||||
_, dst1, _ := net.ParseCIDR("128.0.0.0/1")
|
||||
rt1 := netlink.Route{
|
||||
LinkIndex: p.link.Attrs().Index,
|
||||
Dst: dst1,
|
||||
Gw: gw,
|
||||
Priority: rt.Metric,
|
||||
}
|
||||
p.out.Debug("Point.DelBypass: %s", rt1)
|
||||
if err := netlink.RouteDel(&rt1); err != nil {
|
||||
p.out.Warn("Point.DelBypass: %s %s", rt1.Dst, err)
|
||||
}
|
||||
p.out.Info("Point.DelBypass: route %s via %s", rt1.Dst, rt1.Gw)
|
||||
}
|
||||
}
|
||||
|
||||
func (p *Point) DelRoutes(routes []*models.Route) error {
|
||||
if routes == nil || p.link == nil {
|
||||
return nil
|
||||
}
|
||||
p.DelBypass(routes)
|
||||
for _, rt := range routes {
|
||||
_, dst, err := net.ParseCIDR(rt.Prefix)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
nxt := net.ParseIP(rt.NextHop)
|
||||
rte := netlink.Route{
|
||||
LinkIndex: p.link.Attrs().Index,
|
||||
Dst: dst,
|
||||
Gw: nxt,
|
||||
Priority: rt.Metric,
|
||||
}
|
||||
if err := netlink.RouteDel(&rte); err != nil {
|
||||
p.out.Warn("Point.DelRoute: %s %s", rt.Prefix, err)
|
||||
continue
|
||||
}
|
||||
p.out.Info("Point.DelRoutes: route %s via %s", rt.Prefix, rt.NextHop)
|
||||
}
|
||||
p.routes = nil
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user