func printTypes(md toml.MetaData) {
tabw := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0)
for _, key := range md.Keys() {
fmt.Fprintf(tabw, "%s%s\t%s\n",
strings.Repeat(" ", len(key)-1), key, md.Type(key...))
}
tabw.Flush()
}
func parseConfig(configFile string) map[string]*Service {
var config Services
var md toml.MetaData
var err error
if md, err = toml.DecodeFile(configFile, &config); err != nil {
log.Fatal(err)
}
services := make(map[string]*Service)
for name, serv := range config {
s := NewService()
s.Name = name
services[name] = s
for _, k := range md.Keys() {
ks := strings.Split(k.String(), ".")
if len(ks) > 1 && ks[0] == name {
switch ks[1] {
case "srcport":
s.Srcport = serv.Srcport
case "srcflags":
s.Srcflags = serv.Srcflags
case "srcproto":
s.Srcproto = serv.Srcproto
case "dsthost":
s.Dsthost = serv.Dsthost
case "dstproto":
s.Dstproto = serv.Dstproto
case "dstport":
s.Dstport = serv.Dstport
case "enabled":
s.Enabled = serv.Enabled
}
}
}
}
log.Println("Found", len(services), "services")
return services
}
func SetDefaultConfigValues(md toml.MetaData, bt *Bgp) error {
neighbors := []neighbor{}
global := make(map[string]bool)
for _, key := range md.Keys() {
if !strings.HasPrefix(key.String(), "Global") {
continue
}
if key.String() != "Global" {
global[key.String()] = true
}
}
if _, ok := global["Global.AfiSafis.AfiSafiList"]; !ok {
bt.Global.AfiSafis.AfiSafiList = []AfiSafi{
AfiSafi{AfiSafiName: "ipv4-unicast"},
AfiSafi{AfiSafiName: "ipv6-unicast"},
AfiSafi{AfiSafiName: "l3vpn-ipv4-unicast"},
AfiSafi{AfiSafiName: "l3vpn-ipv6-unicast"},
AfiSafi{AfiSafiName: "l2vpn-evpn"},
AfiSafi{AfiSafiName: "encap"},
AfiSafi{AfiSafiName: "rtc"},
AfiSafi{AfiSafiName: "ipv4-flowspec"},
AfiSafi{AfiSafiName: "l3vpn-ipnv4-flowspec"},
}
}
nidx := 0
for _, key := range md.Keys() {
if !strings.HasPrefix(key.String(), "Neighbors.NeighborList") {
continue
}
if key.String() == "Neighbors.NeighborList" {
neighbors = append(neighbors, neighbor{attributes: make(map[string]bool)})
nidx++
} else {
neighbors[nidx-1].attributes[key.String()] = true
}
}
for i, n := range neighbors {
neighbor := &bt.Neighbors.NeighborList[i]
timerConfig := &neighbor.Timers.TimersConfig
if _, ok := n.attributes["Neighbors.NeighborList.Timers.TimersConfig.ConnectRetry"]; !ok {
timerConfig.HoldTime = float64(DEFAULT_CONNECT_RETRY)
}
if _, ok := n.attributes["Neighbors.NeighborList.Timers.TimersConfig.HoldTime"]; !ok {
timerConfig.HoldTime = float64(DEFAULT_HOLDTIME)
}
if _, ok := n.attributes["Neighbors.NeighborList.Timers.TimersConfig.KeepaliveInterval"]; !ok {
timerConfig.KeepaliveInterval = timerConfig.HoldTime / 3
}
if _, ok := n.attributes["Neighbors.NeighborList.Timers.TimersConfig.IdleHoldTimeAfterReset"]; !ok {
timerConfig.IdleHoldTimeAfterReset = float64(DEFAULT_IDLE_HOLDTIME_AFTER_RESET)
}
if _, ok := n.attributes["Neighbors.NeighborList.AfiSafis.AfiSafiList"]; !ok {
if neighbor.NeighborConfig.NeighborAddress.To4() != nil {
neighbor.AfiSafis.AfiSafiList = []AfiSafi{
AfiSafi{AfiSafiName: "ipv4-unicast"}}
} else {
neighbor.AfiSafis.AfiSafiList = []AfiSafi{
AfiSafi{AfiSafiName: "ipv6-unicast"}}
}
} else {
for _, rf := range neighbor.AfiSafis.AfiSafiList {
_, err := bgp.GetRouteFamily(rf.AfiSafiName)
if err != nil {
return err
}
}
}
if _, ok := n.attributes["Neighbors.NeighborList.NeighborConfig.PeerType"]; !ok {
if neighbor.NeighborConfig.PeerAs != bt.Global.GlobalConfig.As {
neighbor.NeighborConfig.PeerType = PEER_TYPE_EXTERNAL
} else {
neighbor.NeighborConfig.PeerType = PEER_TYPE_INTERNAL
}
}
}
for _, r := range bt.RpkiServers.RpkiServerList {
if r.RpkiServerConfig.Port == 0 {
r.RpkiServerConfig.Port = bgp.RPKI_DEFAULT_PORT
}
}
return nil
}
