mirror of
https://github.com/cloudnativelabs/kube-router.git
synced 2025-10-09 17:01:30 +02:00
286 lines
8.2 KiB
Go
286 lines
8.2 KiB
Go
// Copyright (C) 2016 Nippon Telegraph and Telephone Corporation.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
|
// implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package server
|
|
|
|
import (
|
|
log "github.com/Sirupsen/logrus"
|
|
"github.com/osrg/gobgp/config"
|
|
"github.com/osrg/gobgp/packet/bgp"
|
|
"github.com/osrg/gobgp/table"
|
|
"github.com/stretchr/testify/assert"
|
|
"net"
|
|
"runtime"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestModPolicyAssign(t *testing.T) {
|
|
assert := assert.New(t)
|
|
s := NewBgpServer()
|
|
go s.Serve()
|
|
s.Start(&config.Global{
|
|
Config: config.GlobalConfig{
|
|
As: 1,
|
|
RouterId: "1.1.1.1",
|
|
},
|
|
})
|
|
err := s.AddPolicy(&table.Policy{Name: "p1"}, false)
|
|
assert.Nil(err)
|
|
|
|
err = s.AddPolicy(&table.Policy{Name: "p2"}, false)
|
|
assert.Nil(err)
|
|
|
|
err = s.AddPolicy(&table.Policy{Name: "p3"}, false)
|
|
assert.Nil(err)
|
|
|
|
err = s.AddPolicyAssignment("", table.POLICY_DIRECTION_IMPORT,
|
|
[]*config.PolicyDefinition{&config.PolicyDefinition{Name: "p1"}, &config.PolicyDefinition{Name: "p2"}, &config.PolicyDefinition{Name: "p3"}}, table.ROUTE_TYPE_ACCEPT)
|
|
assert.Nil(err)
|
|
|
|
err = s.DeletePolicyAssignment("", table.POLICY_DIRECTION_IMPORT,
|
|
[]*config.PolicyDefinition{&config.PolicyDefinition{Name: "p1"}}, false)
|
|
assert.Nil(err)
|
|
|
|
_, ps, _ := s.GetPolicyAssignment("", table.POLICY_DIRECTION_IMPORT)
|
|
assert.Equal(len(ps), 2)
|
|
}
|
|
|
|
func TestMonitor(test *testing.T) {
|
|
assert := assert.New(test)
|
|
s := NewBgpServer()
|
|
go s.Serve()
|
|
s.Start(&config.Global{
|
|
Config: config.GlobalConfig{
|
|
As: 1,
|
|
RouterId: "1.1.1.1",
|
|
Port: 10179,
|
|
},
|
|
})
|
|
n := &config.Neighbor{
|
|
Config: config.NeighborConfig{
|
|
NeighborAddress: "127.0.0.1",
|
|
PeerAs: 2,
|
|
},
|
|
Transport: config.Transport{
|
|
Config: config.TransportConfig{
|
|
PassiveMode: true,
|
|
},
|
|
},
|
|
}
|
|
if err := s.AddNeighbor(n); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
t := NewBgpServer()
|
|
go t.Serve()
|
|
t.Start(&config.Global{
|
|
Config: config.GlobalConfig{
|
|
As: 2,
|
|
RouterId: "2.2.2.2",
|
|
Port: -1,
|
|
},
|
|
})
|
|
m := &config.Neighbor{
|
|
Config: config.NeighborConfig{
|
|
NeighborAddress: "127.0.0.1",
|
|
PeerAs: 1,
|
|
},
|
|
Transport: config.Transport{
|
|
Config: config.TransportConfig{
|
|
RemotePort: 10179,
|
|
},
|
|
},
|
|
}
|
|
if err := t.AddNeighbor(m); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
for {
|
|
time.Sleep(time.Second)
|
|
if t.GetNeighbor(false)[0].State.SessionState == config.SESSION_STATE_ESTABLISHED {
|
|
break
|
|
}
|
|
}
|
|
|
|
w := s.Watch(WatchBestPath(false))
|
|
|
|
attrs := []bgp.PathAttributeInterface{
|
|
bgp.NewPathAttributeOrigin(0),
|
|
bgp.NewPathAttributeNextHop("10.0.0.1"),
|
|
}
|
|
if _, err := t.AddPath("", []*table.Path{table.NewPath(nil, bgp.NewIPAddrPrefix(24, "10.0.0.0"), false, attrs, time.Now(), false)}); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
ev := <-w.Event()
|
|
b := ev.(*WatchEventBestPath)
|
|
assert.Equal(len(b.PathList), 1)
|
|
assert.Equal(b.PathList[0].GetNlri().String(), "10.0.0.0/24")
|
|
assert.Equal(b.PathList[0].IsWithdraw, false)
|
|
|
|
if _, err := t.AddPath("", []*table.Path{table.NewPath(nil, bgp.NewIPAddrPrefix(24, "10.0.0.0"), true, attrs, time.Now(), false)}); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
ev = <-w.Event()
|
|
b = ev.(*WatchEventBestPath)
|
|
assert.Equal(len(b.PathList), 1)
|
|
assert.Equal(b.PathList[0].GetNlri().String(), "10.0.0.0/24")
|
|
assert.Equal(b.PathList[0].IsWithdraw, true)
|
|
|
|
if _, err := t.AddPath("", []*table.Path{table.NewPath(nil, bgp.NewIPAddrPrefix(24, "10.0.0.0"), true, attrs, time.Now(), false)}); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
//stop the watcher still having an item.
|
|
w.Stop()
|
|
}
|
|
|
|
func TestNumGoroutineWithAddDeleteNeighbor(t *testing.T) {
|
|
assert := assert.New(t)
|
|
s := NewBgpServer()
|
|
go s.Serve()
|
|
err := s.Start(&config.Global{
|
|
Config: config.GlobalConfig{
|
|
As: 1,
|
|
RouterId: "1.1.1.1",
|
|
Port: -1,
|
|
},
|
|
})
|
|
assert.Nil(err)
|
|
|
|
num := runtime.NumGoroutine()
|
|
|
|
n := &config.Neighbor{
|
|
Config: config.NeighborConfig{
|
|
NeighborAddress: "127.0.0.1",
|
|
PeerAs: 2,
|
|
},
|
|
Transport: config.Transport{
|
|
Config: config.TransportConfig{
|
|
PassiveMode: true,
|
|
},
|
|
},
|
|
}
|
|
err = s.AddNeighbor(n)
|
|
assert.Nil(err)
|
|
|
|
err = s.DeleteNeighbor(n)
|
|
assert.Nil(err)
|
|
// wait goroutines to finish (e.g. internal goroutine for
|
|
// InfiniteChannel)
|
|
time.Sleep(time.Second * 5)
|
|
assert.Equal(num, runtime.NumGoroutine())
|
|
}
|
|
|
|
func newPeerandInfo(myAs, as uint32, address string, rib *table.TableManager) (*Peer, *table.PeerInfo) {
|
|
nConf := &config.Neighbor{Config: config.NeighborConfig{PeerAs: as, NeighborAddress: address}}
|
|
config.SetDefaultNeighborConfigValues(nConf, myAs)
|
|
p := NewPeer(
|
|
&config.Global{Config: config.GlobalConfig{As: myAs}},
|
|
nConf,
|
|
rib,
|
|
&table.RoutingPolicy{})
|
|
for _, f := range rib.GetRFlist() {
|
|
p.fsm.rfMap[f] = true
|
|
}
|
|
return p, &table.PeerInfo{AS: as, Address: net.ParseIP(address)}
|
|
}
|
|
|
|
func process(rib *table.TableManager, l []*table.Path) (*table.Path, *table.Path) {
|
|
news, olds, _ := rib.ProcessPaths([]string{table.GLOBAL_RIB_NAME}, l)
|
|
if len(news) != 1 {
|
|
panic("can't handle multiple paths")
|
|
}
|
|
for idx, path := range news[table.GLOBAL_RIB_NAME] {
|
|
var old *table.Path
|
|
if olds != nil {
|
|
old = olds[table.GLOBAL_RIB_NAME][idx]
|
|
}
|
|
return path, old
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func TestFilterpathWitheBGP(t *testing.T) {
|
|
as := uint32(65000)
|
|
p1As := uint32(65001)
|
|
p2As := uint32(65002)
|
|
rib := table.NewTableManager([]bgp.RouteFamily{bgp.RF_IPv4_UC})
|
|
p1, pi1 := newPeerandInfo(as, p1As, "192.168.0.1", rib)
|
|
p2, pi2 := newPeerandInfo(as, p2As, "192.168.0.2", rib)
|
|
|
|
nlri := bgp.NewIPAddrPrefix(24, "10.10.10.0")
|
|
pa1 := []bgp.PathAttributeInterface{bgp.NewPathAttributeAsPath([]bgp.AsPathParamInterface{bgp.NewAs4PathParam(2, []uint32{p1As})}), bgp.NewPathAttributeLocalPref(200)}
|
|
pa2 := []bgp.PathAttributeInterface{bgp.NewPathAttributeAsPath([]bgp.AsPathParamInterface{bgp.NewAs4PathParam(2, []uint32{p2As})})}
|
|
|
|
path1 := table.NewPath(pi1, nlri, false, pa1, time.Now(), false)
|
|
path2 := table.NewPath(pi2, nlri, false, pa2, time.Now(), false)
|
|
|
|
new, old := process(rib, []*table.Path{path1, path2})
|
|
assert.Equal(t, new, path1)
|
|
filterpath(p1, new, old)
|
|
filterpath(p2, new, old)
|
|
|
|
new, old = process(rib, []*table.Path{path1.Clone(true)})
|
|
assert.Equal(t, new, path2)
|
|
// p1 and p2 advertized the same prefix and p1's was best. Then p1 withdraw it, so p2 must get withdawal.
|
|
path := filterpath(p2, new, old)
|
|
assert.NotNil(t, path)
|
|
assert.True(t, path.IsWithdraw)
|
|
|
|
// p1 should get the new best (from p2)
|
|
assert.Equal(t, filterpath(p1, new, old), path2)
|
|
|
|
new, old = process(rib, []*table.Path{path2.Clone(true)})
|
|
assert.True(t, new.IsWithdraw)
|
|
// p2 withdraw so p1 should get withdrawal.
|
|
path = filterpath(p1, new, old)
|
|
assert.True(t, path.IsWithdraw)
|
|
|
|
// p2 withdraw so p2 should get nothing.
|
|
path = filterpath(p2, new, old)
|
|
assert.Nil(t, path)
|
|
}
|
|
|
|
func TestFilterpathWithiBGP(t *testing.T) {
|
|
as := uint32(65000)
|
|
|
|
rib := table.NewTableManager([]bgp.RouteFamily{bgp.RF_IPv4_UC})
|
|
p1, pi1 := newPeerandInfo(as, as, "192.168.0.1", rib)
|
|
//p2, pi2 := newPeerandInfo(as, as, "192.168.0.2", rib)
|
|
p2, _ := newPeerandInfo(as, as, "192.168.0.2", rib)
|
|
|
|
nlri := bgp.NewIPAddrPrefix(24, "10.10.10.0")
|
|
pa1 := []bgp.PathAttributeInterface{bgp.NewPathAttributeAsPath([]bgp.AsPathParamInterface{bgp.NewAs4PathParam(2, []uint32{as})}), bgp.NewPathAttributeLocalPref(200)}
|
|
//pa2 := []bgp.PathAttributeInterface{bgp.NewPathAttributeAsPath([]bgp.AsPathParamInterface{bgp.NewAs4PathParam(2, []uint32{as})})}
|
|
|
|
path1 := table.NewPath(pi1, nlri, false, pa1, time.Now(), false)
|
|
//path2 := table.NewPath(pi2, nlri, false, pa2, time.Now(), false)
|
|
|
|
new, old := process(rib, []*table.Path{path1})
|
|
assert.Equal(t, new, path1)
|
|
path := filterpath(p1, new, old)
|
|
assert.Nil(t, path)
|
|
path = filterpath(p2, new, old)
|
|
assert.Nil(t, path)
|
|
|
|
new, old = process(rib, []*table.Path{path1.Clone(true)})
|
|
path = filterpath(p1, new, old)
|
|
assert.Nil(t, path)
|
|
path = filterpath(p2, new, old)
|
|
assert.Nil(t, path)
|
|
|
|
}
|