2020-04-12 08:55:13 -04:00
|
|
|
package routing
|
|
|
|
|
|
|
|
|
|
import (
|
2020-04-19 18:13:48 +00:00
|
|
|
"context"
|
2020-04-12 08:55:13 -04:00
|
|
|
"fmt"
|
2020-10-20 02:45:28 +00:00
|
|
|
"net"
|
2020-04-12 08:55:13 -04:00
|
|
|
)
|
|
|
|
|
|
2020-10-20 02:45:28 +00:00
|
|
|
func (r *routing) AddRouteVia(ctx context.Context,
|
|
|
|
|
subnet net.IPNet, defaultGateway net.IP, defaultInterface string) error {
|
2020-07-11 21:03:55 +00:00
|
|
|
subnetStr := subnet.String()
|
|
|
|
|
r.logger.Info("adding %s as route via %s %s", subnetStr, defaultGateway, defaultInterface)
|
|
|
|
|
exists, err := r.routeExists(subnet)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return err
|
|
|
|
|
} else if exists {
|
|
|
|
|
return nil
|
|
|
|
|
}
|
2020-07-13 02:14:56 +00:00
|
|
|
if r.debug {
|
|
|
|
|
fmt.Printf("ip route add %s via %s dev %s\n", subnetStr, defaultGateway, defaultInterface)
|
|
|
|
|
}
|
2020-10-20 02:45:28 +00:00
|
|
|
output, err := r.commander.Run(ctx,
|
|
|
|
|
"ip", "route", "add", subnetStr, "via", defaultGateway.String(), "dev", defaultInterface)
|
2020-07-11 21:03:55 +00:00
|
|
|
if err != nil {
|
2020-10-20 02:45:28 +00:00
|
|
|
return fmt.Errorf("cannot add route for %s via %s %s %s: %s: %w",
|
|
|
|
|
subnetStr, defaultGateway, "dev", defaultInterface, output, err)
|
2020-04-12 08:55:13 -04:00
|
|
|
}
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
2020-07-11 21:03:55 +00:00
|
|
|
func (r *routing) DeleteRouteVia(ctx context.Context, subnet net.IPNet) (err error) {
|
|
|
|
|
subnetStr := subnet.String()
|
|
|
|
|
r.logger.Info("deleting route for %s", subnetStr)
|
|
|
|
|
exists, err := r.routeExists(subnet)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return err
|
|
|
|
|
} else if !exists { // thanks to @npawelek https://github.com/npawelek
|
|
|
|
|
return nil
|
|
|
|
|
}
|
2020-07-13 02:14:56 +00:00
|
|
|
if r.debug {
|
|
|
|
|
fmt.Printf("ip route del %s\n", subnetStr)
|
|
|
|
|
}
|
2020-07-11 21:03:55 +00:00
|
|
|
output, err := r.commander.Run(ctx, "ip", "route", "del", subnetStr)
|
2020-04-12 08:55:13 -04:00
|
|
|
if err != nil {
|
2020-07-11 21:03:55 +00:00
|
|
|
return fmt.Errorf("cannot delete route for %s: %s: %w", subnetStr, output, err)
|
2020-04-12 08:55:13 -04:00
|
|
|
}
|
|
|
|
|
return nil
|
|
|
|
|
}
|