36 lines
1.1 KiB
Go
36 lines
1.1 KiB
Go
package routing
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
|
|
"fmt"
|
|
)
|
|
|
|
func (r *routing) AddRoutesVia(ctx context.Context, subnets []net.IPNet, defaultGateway net.IP, defaultInterface string) error {
|
|
for _, subnet := range subnets {
|
|
exists, err := r.routeExists(subnet)
|
|
if err != nil {
|
|
return err
|
|
} else if exists { // thanks to @npawelek https://github.com/npawelek
|
|
if err := r.removeRoute(ctx, subnet); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
r.logger.Info("adding %s as route via %s", subnet.String(), defaultInterface)
|
|
output, err := r.commander.Run(ctx, "ip", "route", "add", subnet.String(), "via", defaultGateway.String(), "dev", defaultInterface)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot add route for %s via %s %s %s: %s: %w", subnet.String(), defaultGateway.String(), "dev", defaultInterface, output, err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *routing) removeRoute(ctx context.Context, subnet net.IPNet) (err error) {
|
|
output, err := r.commander.Run(ctx, "ip", "route", "del", subnet.String())
|
|
if err != nil {
|
|
return fmt.Errorf("cannot delete route for %s: %s: %w", subnet.String(), output, err)
|
|
}
|
|
return nil
|
|
}
|