route adding
This commit is contained in:
parent
04c2cebba6
commit
81d061a316
39
group.go
39
group.go
@ -9,12 +9,14 @@ import (
|
|||||||
"kvas2-go/models"
|
"kvas2-go/models"
|
||||||
"kvas2-go/pkg/ip-helper"
|
"kvas2-go/pkg/ip-helper"
|
||||||
|
|
||||||
|
"github.com/rs/zerolog/log"
|
||||||
"github.com/vishvananda/netlink"
|
"github.com/vishvananda/netlink"
|
||||||
)
|
)
|
||||||
|
|
||||||
type GroupOptions struct {
|
type GroupOptions struct {
|
||||||
Enabled bool
|
Enabled bool
|
||||||
ipRule *netlink.Rule
|
ipRule *netlink.Rule
|
||||||
|
ipRoute *netlink.Route
|
||||||
}
|
}
|
||||||
|
|
||||||
type Group struct {
|
type Group struct {
|
||||||
@ -75,6 +77,27 @@ func (g *Group) Enable() error {
|
|||||||
}
|
}
|
||||||
g.options.ipRule = rule
|
g.options.ipRule = rule
|
||||||
|
|
||||||
|
iface, err := netlink.LinkByName(g.Interface)
|
||||||
|
if err != nil {
|
||||||
|
log.Warn().Str("interface", g.Interface).Msg("error while getting interface")
|
||||||
|
}
|
||||||
|
|
||||||
|
if iface != nil {
|
||||||
|
route := &netlink.Route{
|
||||||
|
LinkIndex: iface.Attrs().Index,
|
||||||
|
Table: rule.Table,
|
||||||
|
Dst: &net.IPNet{
|
||||||
|
IP: []byte{0, 0, 0, 0},
|
||||||
|
Mask: []byte{0, 0, 0, 0},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
err = netlink.RouteAdd(route)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("error while adding route: %w", err)
|
||||||
|
}
|
||||||
|
g.options.ipRoute = route
|
||||||
|
}
|
||||||
|
|
||||||
defaultTimeout := uint32(300)
|
defaultTimeout := uint32(300)
|
||||||
err = netlink.IpsetDestroy(g.ipsetName)
|
err = netlink.IpsetDestroy(g.ipsetName)
|
||||||
if err != nil && !os.IsNotExist(err) {
|
if err != nil && !os.IsNotExist(err) {
|
||||||
@ -97,10 +120,23 @@ func (g *Group) Disable() error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
err := netlink.RuleDel(g.options.ipRule)
|
var err error
|
||||||
|
|
||||||
|
if g.options.ipRule != nil {
|
||||||
|
err = netlink.RuleDel(g.options.ipRule)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("error while deleting rule: %w", err)
|
return fmt.Errorf("error while deleting rule: %w", err)
|
||||||
}
|
}
|
||||||
|
g.options.ipRule = nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if g.options.ipRoute != nil {
|
||||||
|
err = netlink.RouteDel(g.options.ipRoute)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("error while deleting route: %w", err)
|
||||||
|
}
|
||||||
|
g.options.ipRule = nil
|
||||||
|
}
|
||||||
|
|
||||||
err = netlink.IpsetDestroy(g.ipsetName)
|
err = netlink.IpsetDestroy(g.ipsetName)
|
||||||
if err != nil && !os.IsNotExist(err) {
|
if err != nil && !os.IsNotExist(err) {
|
||||||
@ -108,7 +144,6 @@ func (g *Group) Disable() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
g.options.Enabled = false
|
g.options.Enabled = false
|
||||||
g.options.ipRule = nil
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user