policy: support large-community match/action

close #1133

Signed-off-by: Wataru Ishida <ishida.wataru@lab.ntt.co.jp>

1 files changed, 80 insertions, 13 deletions

diff --git a/table/policy_test.go b/table/policy_test.go
index cfafb1ed..58b944e2 100644
--- a/table/policy_test.go
+++ b/table/policy_test.go
@@ -29,12 +29,7 @@ import (
 	"time"
 )
 
-func init() {
-	log.SetLevel(log.DebugLevel)
-}
-
 func TestPrefixCalcurateNoRange(t *testing.T) {
-	log.SetLevel(log.DebugLevel)
 	// create path
 	peer := &PeerInfo{AS: 65001, Address: net.ParseIP("10.0.0.1")}
 	origin := bgp.NewPathAttributeOrigin(0)
@@ -125,7 +120,6 @@ func TestPrefixCalcurateLengthRange(t *testing.T) {
 }
 
 func TestPrefixCalcurateNoRangeIPv6(t *testing.T) {
-	log.SetLevel(log.DebugLevel)
 	// create path
 	peer := &PeerInfo{AS: 65001, Address: net.ParseIP("2001::192:168:50:1")}
 	origin := bgp.NewPathAttributeOrigin(0)
@@ -1415,8 +1409,6 @@ func TestAs4PathConditionEvaluateMixedWith2byteAS(t *testing.T) {
 
 func TestCommunityConditionEvaluate(t *testing.T) {
 
-	log.SetLevel(log.DebugLevel)
-
 	// setup
 	// create path
 	peer := &PeerInfo{AS: 65001, Address: net.ParseIP("10.0.0.1")}
@@ -1681,7 +1673,6 @@ func TestPolicyMatchAndAddCommunities(t *testing.T) {
 	pType, newPath := p.Apply(path, nil)
 	assert.Equal(t, ROUTE_TYPE_ACCEPT, pType)
 	assert.NotEqual(t, nil, newPath)
-	log.Debug(newPath)
 	assert.Equal(t, []uint32{stringToCommunityValue(community)}, newPath.GetCommunities())
 }
 
@@ -1914,8 +1905,6 @@ func TestPolicyMatchAndClearCommunities(t *testing.T) {
 
 func TestExtCommunityConditionEvaluate(t *testing.T) {
 
-	log.SetLevel(log.DebugLevel)
-
 	// setup
 	// create path
 	peer := &PeerInfo{AS: 65001, Address: net.ParseIP("10.0.0.1")}
@@ -2097,8 +2086,6 @@ func TestExtCommunityConditionEvaluate(t *testing.T) {
 
 func TestExtCommunityConditionEvaluateWithOtherCondition(t *testing.T) {
 
-	log.SetLevel(log.DebugLevel)
-
 	// setup
 	// create path
 	peer := &PeerInfo{AS: 65001, Address: net.ParseIP("10.2.1.1")}
@@ -2871,5 +2858,85 @@ func TestPrefixSetMatch(t *testing.T) {
 
 	path = NewPath(nil, bgp.NewIPAddrPrefix(6, "0.0.0.0"), false, []bgp.PathAttributeInterface{}, time.Now(), false)
 	assert.False(t, m.Evaluate(path, nil))
+}
+
+func TestLargeCommunityMatchAction(t *testing.T) {
+	coms := []*bgp.LargeCommunity{
+		&bgp.LargeCommunity{100, 100, 100},
+		&bgp.LargeCommunity{100, 200, 200},
+	}
+	p := NewPath(nil, nil, false, []bgp.PathAttributeInterface{bgp.NewPathAttributeLargeCommunities(coms)}, time.Time{}, false)
+
+	c := config.LargeCommunitySet{
+		LargeCommunitySetName: "l0",
+		LargeCommunityList: []string{
+			"100:100:100",
+			"100:300:100",
+		},
+	}
+
+	set, err := NewLargeCommunitySet(c)
+	assert.Equal(t, err, nil)
+
+	m, err := NewLargeCommunityCondition(config.MatchLargeCommunitySet{
+		LargeCommunitySet: "l0",
+	})
+	assert.Equal(t, err, nil)
+	m.set = set
+
+	assert.Equal(t, m.Evaluate(p, nil), true)
+
+	a, err := NewLargeCommunityAction(config.SetLargeCommunity{
+		config.SetLargeCommunityMethod{
+			[]string{"100:100:100"},
+		},
+		config.BGP_SET_COMMUNITY_OPTION_TYPE_REMOVE,
+	})
+	assert.Equal(t, err, nil)
+	p = a.Apply(p, nil)
+
+	assert.Equal(t, m.Evaluate(p, nil), false)
+
+	a, err = NewLargeCommunityAction(config.SetLargeCommunity{
+		config.SetLargeCommunityMethod{
+			[]string{
+				"100:300:100",
+				"200:100:100",
+			},
+		},
+		config.BGP_SET_COMMUNITY_OPTION_TYPE_ADD,
+	})
+	assert.Equal(t, err, nil)
+	p = a.Apply(p, nil)
+
+	assert.Equal(t, m.Evaluate(p, nil), true)
+
+	a, err = NewLargeCommunityAction(config.SetLargeCommunity{
+		config.SetLargeCommunityMethod{
+			[]string{"^100:"},
+		},
+		config.BGP_SET_COMMUNITY_OPTION_TYPE_REMOVE,
+	})
+	assert.Equal(t, err, nil)
+	p = a.Apply(p, nil)
+
+	assert.Equal(t, m.Evaluate(p, nil), false)
+
+	c = config.LargeCommunitySet{
+		LargeCommunitySetName: "l1",
+		LargeCommunityList: []string{
+			"200:",
+		},
+	}
+
+	set, err = NewLargeCommunitySet(c)
+	assert.Equal(t, err, nil)
+
+	m, err = NewLargeCommunityCondition(config.MatchLargeCommunitySet{
+		LargeCommunitySet: "l1",
+	})
+	assert.Equal(t, err, nil)
+	m.set = set
 
+	assert.Equal(t, m.Evaluate(p, nil), true)
 }