summaryrefslogtreecommitdiffhomepage
path: root/test/scenario_test/quagga-rsconfig.go
diff options
context:
space:
mode:
Diffstat (limited to 'test/scenario_test/quagga-rsconfig.go')
-rw-r--r--test/scenario_test/quagga-rsconfig.go211
1 files changed, 207 insertions, 4 deletions
diff --git a/test/scenario_test/quagga-rsconfig.go b/test/scenario_test/quagga-rsconfig.go
index 9f9592d3..2f64ce7b 100644
--- a/test/scenario_test/quagga-rsconfig.go
+++ b/test/scenario_test/quagga-rsconfig.go
@@ -88,7 +88,7 @@ func (qt *QuaggaConfig) IPv6Config() *bytes.Buffer {
return buf
}
-func create_config_files(nr int, outputDir string, IPVersion string, nonePeer bool, normalBGP bool) {
+func create_config_files(nr int, outputDir string, IPVersion string, nonePeer bool, normalBGP bool, policyPattern string) {
quaggaConfigList := make([]*QuaggaConfig, 0)
gobgpConf := config.Bgp{
@@ -98,6 +98,11 @@ func create_config_files(nr int, outputDir string, IPVersion string, nonePeer bo
},
}
+ var binding *PolicyBinding
+ if policyPattern != "" {
+ binding = bindPolicy(policyPattern)
+ }
+
for i := 1; i < nr+1; i++ {
c := config.Neighbor{
PeerAs: 65000 + uint32(i),
@@ -108,6 +113,19 @@ func create_config_files(nr int, outputDir string, IPVersion string, nonePeer bo
Timers: config.Timers{HoldTime: 30, KeepaliveInterval: 10, IdleHoldTimeAfterReset: 10},
PeerType: config.PEER_TYPE_EXTERNAL,
}
+
+ if binding != nil {
+
+ if binding.NeighborAddress == c.NeighborAddress.String() {
+ ap := config.ApplyPolicy{
+ ImportPolicies: binding.ImportPolicyNames,
+ ExportPolicies: binding.ExportPolicyNames,
+ }
+ c.ApplyPolicy = ap
+ }
+
+ }
+
gobgpConf.NeighborList = append(gobgpConf.NeighborList, c)
if !nonePeer {
q := NewQuaggaConfig(i, &gobgpConf.Global, &c, net.ParseIP(serverAddress[IPVersion]))
@@ -128,6 +146,9 @@ func create_config_files(nr int, outputDir string, IPVersion string, nonePeer bo
var buffer bytes.Buffer
encoder := toml.NewEncoder(&buffer)
encoder.Encode(gobgpConf)
+ if binding != nil {
+ encoder.Encode(binding.Policy)
+ }
err := ioutil.WriteFile(fmt.Sprintf("%s/gobgpd.conf", outputDir), buffer.Bytes(), 0644)
if err != nil {
@@ -180,6 +201,180 @@ func append_config_files(ar int, outputDir string, IPVersion string, nonePeer bo
}
}
+func createPolicyConfig() *config.RoutingPolicy {
+
+ ps0 := config.PrefixSet{
+ PrefixSetName: "ps0",
+ PrefixList: []config.Prefix{
+ config.Prefix{
+ Address: net.ParseIP("192.168.0.0"),
+ Masklength: 16,
+ MasklengthRange: "16..24",
+ }},
+ }
+
+ ps1 := config.PrefixSet{
+ PrefixSetName: "ps1",
+ PrefixList: []config.Prefix{
+ config.Prefix{
+ Address: net.ParseIP("192.168.20.0"),
+ Masklength: 24,
+ }, config.Prefix{
+ Address: net.ParseIP("192.168.200.0"),
+ Masklength: 24,
+ }},
+ }
+
+ ps2 := config.PrefixSet{
+ PrefixSetName: "ps2",
+ PrefixList: []config.Prefix{
+ config.Prefix{
+ Address: net.ParseIP("192.168.20.0"),
+ Masklength: 24,
+ }},
+ }
+
+ ns0 := config.NeighborSet{
+ NeighborSetName: "ns0",
+ NeighborInfoList: []config.NeighborInfo{
+ config.NeighborInfo{
+ Address: net.ParseIP("10.0.0.2"),
+ }},
+ }
+
+ ds := config.DefinedSets{
+ PrefixSetList: []config.PrefixSet{ps0, ps1, ps2},
+ NeighborSetList: []config.NeighborSet{ns0},
+ }
+
+ st0 := config.Statement{
+ Name: "st0",
+ Conditions: config.Conditions{
+ MatchPrefixSet: "ps0",
+ MatchNeighborSet: "ns0",
+ MatchSetOptions: config.MATCH_SET_OPTIONS_TYPE_ALL,
+ },
+ Actions: config.Actions{
+ AcceptRoute: false,
+ RejectRoute: true,
+ },
+ }
+
+ st1 := config.Statement{
+ Name: "st1",
+ Conditions: config.Conditions{
+ MatchPrefixSet: "ps1",
+ MatchNeighborSet: "ns0",
+ MatchSetOptions: config.MATCH_SET_OPTIONS_TYPE_ALL,
+ },
+ Actions: config.Actions{
+ AcceptRoute: false,
+ RejectRoute: true,
+ },
+ }
+
+ pd0 := config.PolicyDefinition{
+ Name: "policy0",
+ StatementList: []config.Statement{st0},
+ }
+
+ pd1 := config.PolicyDefinition{
+ Name: "policy1",
+ StatementList: []config.Statement{st1},
+ }
+
+ p := &config.RoutingPolicy{
+ DefinedSets: ds,
+ PolicyDefinitionList: []config.PolicyDefinition{pd0, pd1},
+ }
+ return p
+}
+
+func updatePolicyConfig(outputDir string, pattern string) {
+
+ newConf := config.Bgp{}
+ policyConf := config.RoutingPolicy{}
+
+ _, d_err := toml.DecodeFile(fmt.Sprintf("%s/gobgpd.conf", outputDir), &newConf)
+ if d_err != nil {
+ log.Fatal(d_err)
+ }
+ _, d_err = toml.DecodeFile(fmt.Sprintf("%s/gobgpd.conf", outputDir), &policyConf)
+ if d_err != nil {
+ log.Fatal(d_err)
+ }
+
+ fmt.Println(policyConf)
+
+ st2 := config.Statement{
+ Name: "st2",
+ Conditions: config.Conditions{
+ MatchPrefixSet: "ps2",
+ MatchNeighborSet: "ns0",
+ MatchSetOptions: config.MATCH_SET_OPTIONS_TYPE_ALL,
+ },
+ Actions: config.Actions{
+ AcceptRoute: false,
+ RejectRoute: true,
+ },
+ }
+
+ if pattern == "p3" || pattern == "p4" {
+ policyConf.PolicyDefinitionList[1].StatementList = []config.Statement{st2}
+ }
+
+ var buffer bytes.Buffer
+ encoder := toml.NewEncoder(&buffer)
+ encoder.Encode(newConf)
+ encoder.Encode(policyConf)
+ e_err := ioutil.WriteFile(fmt.Sprintf("%s/gobgpd.conf", outputDir), buffer.Bytes(), 0644)
+ if e_err != nil {
+ log.Fatal(e_err)
+ }
+
+ return
+}
+
+type PolicyBinding struct {
+ NeighborAddress string
+ Policy *config.RoutingPolicy
+ ImportPolicyNames []string
+ ExportPolicyNames []string
+}
+
+func bindPolicy(pattern string) *PolicyBinding {
+
+ var pb *PolicyBinding = &PolicyBinding{Policy: createPolicyConfig()}
+
+ switch pattern {
+ case "p1":
+ pb.NeighborAddress = "10.0.0.3"
+ pb.ImportPolicyNames = []string{"policy0"}
+ pb.ExportPolicyNames = nil
+
+ case "p2":
+ pb.NeighborAddress = "10.0.0.3"
+ pb.ImportPolicyNames = nil
+ pb.ExportPolicyNames = []string{"policy0"}
+
+ case "p3":
+ pb.NeighborAddress = "10.0.0.3"
+ pb.ImportPolicyNames = []string{"policy1"}
+ pb.ExportPolicyNames = nil
+
+ case "p4":
+ pb.NeighborAddress = "10.0.0.3"
+ pb.ImportPolicyNames = nil
+ pb.ExportPolicyNames = []string{"policy1"}
+
+ default:
+ pb = nil
+ }
+
+ return pb
+
+}
+
func main() {
var opts struct {
ClientNumber int `short:"n" long:"client-number" description:"specfying the number of clients" default:"8"`
@@ -189,6 +384,8 @@ func main() {
NetIdentifier int `short:"i" long:"net-identifer" description:"specifing the use network identifier" default:"0"`
NonePeer bool `long:"none-peer" description:"disable make quagga config"`
NormalBGP bool `long:"normal-bgp" description:"generate normal bgp server configuration"`
+ PolicyPattern string `short:"p" long:"policy-pattern" description:"specify policy pattern" default:""`
+ UpdatePolicy bool `long:"update-policy" description:"update exsisting policy config" default:"false"`
}
parser := flags.NewParser(&opts, flags.Default)
_, err := parser.Parse()
@@ -217,12 +414,18 @@ func main() {
baseNeighborNetMask[IPv4] = ".0/24"
}
- if opts.AppendClient == 0 {
- create_config_files(opts.ClientNumber, opts.OutputDir, opts.IPVersion, opts.NonePeer, opts.NormalBGP)
+ isCreateMode := opts.AppendClient == 0 && !opts.UpdatePolicy
+
+ if isCreateMode {
+ create_config_files(opts.ClientNumber, opts.OutputDir, opts.IPVersion, opts.NonePeer, opts.NormalBGP, opts.PolicyPattern)
} else {
if _, err := os.Stat(fmt.Sprintf("%s/gobgpd.conf", opts.OutputDir)); os.IsNotExist(err) {
log.Fatal(err)
}
- append_config_files(opts.AppendClient, opts.OutputDir, opts.IPVersion, opts.NonePeer, opts.NormalBGP)
+ if opts.UpdatePolicy {
+ updatePolicyConfig(opts.OutputDir, opts.PolicyPattern)
+ } else {
+ append_config_files(opts.AppendClient, opts.OutputDir, opts.IPVersion, opts.NonePeer, opts.NormalBGP)
+ }
}
}
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-15 20:26:51 +0000