84 lines
1.8 KiB
Go
84 lines
1.8 KiB
Go
|
package client
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type IProute struct {
|
||
|
|
Destination string
|
||
|
|
Gateway string
|
||
|
|
Interface string
|
||
|
|
Metric string
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetIpRoutesCmd() string {
|
||
|
|
return "show ip route"
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Client) GetIpRoutes() []IProute {
|
||
|
|
raw := c.ExecuteCommand(GetIpRoutesCmd())
|
||
|
|
|
||
|
|
prefix := 305
|
||
|
|
postfix := 12
|
||
|
|
|
||
|
|
string_result := string(raw[prefix : len(raw)-postfix])
|
||
|
|
lines := strings.Split(string_result, "\r\n")
|
||
|
|
|
||
|
|
result := make([]IProute, len(lines))
|
||
|
|
|
||
|
|
for i, x := range lines {
|
||
|
|
l := strings.Fields(x)
|
||
|
|
result[i] = IProute{Destination: l[0], Gateway: l[1], Interface: l[2], Metric: l[3]}
|
||
|
|
}
|
||
|
|
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func AddIpRouteCmd(route IProute, auto bool) string {
|
||
|
|
autoString := "auto"
|
||
|
|
|
||
|
|
if !auto {
|
||
|
|
autoString = ""
|
||
|
|
}
|
||
|
|
|
||
|
|
return fmt.Sprintf("ip route %s %s %s %s %s", route.Destination, route.Gateway, route.Interface, route.Metric, autoString)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Client) AddIpRoute(route IProute, auto bool) error {
|
||
|
|
cmd := AddIpRouteCmd(route, auto)
|
||
|
|
raw := c.ExecuteCommand(cmd)
|
||
|
|
|
||
|
|
prefix := len(cmd)*4 + 7
|
||
|
|
postfix := 11
|
||
|
|
|
||
|
|
result := string(raw[prefix : len(raw)-postfix])
|
||
|
|
|
||
|
|
if strings.HasPrefix(result, "Network::RoutingTable: added static route") || strings.HasPrefix(result, "Network::RoutingTable: renewed static route") {
|
||
|
|
return nil
|
||
|
|
} else {
|
||
|
|
return errors.New(result)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func RemoveIpRouteCmd(route IProute) string {
|
||
|
|
return fmt.Sprintf("no ip route %s %s %s", route.Destination, route.Interface, route.Metric)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Client) RemoveIpRoute(route IProute) error {
|
||
|
|
cmd := RemoveIpRouteCmd(route)
|
||
|
|
raw := c.ExecuteCommand(cmd)
|
||
|
|
|
||
|
|
prefix := len(cmd)*4 + 7
|
||
|
|
postfix := 11
|
||
|
|
|
||
|
|
result := string(raw[prefix : len(raw)-postfix])
|
||
|
|
|
||
|
|
if strings.HasPrefix(result, "Network::RoutingTable: deleted static route:") {
|
||
|
|
return nil
|
||
|
|
} else {
|
||
|
|
return errors.New(result)
|
||
|
|
}
|
||
|
|
}
|