diff options
Diffstat (limited to 'packet/mrt')
| -rw-r--r-- | packet/mrt/mrt.go | 8 | ||||
| -rw-r--r-- | packet/mrt/mrt_test.go | 16 |
2 files changed, 12 insertions, 12 deletions
diff --git a/packet/mrt/mrt.go b/packet/mrt/mrt.go index 559455d1..1b7c3e05 100644 --- a/packet/mrt/mrt.go +++ b/packet/mrt/mrt.go @@ -431,13 +431,13 @@ func (e *RibEntry) Serialize() ([]byte, error) { return buf, nil } -func NewRibEntry(index uint16, time uint32, pathid uint32, pathattrs []bgp.PathAttributeInterface) *RibEntry { +func NewRibEntry(index uint16, time uint32, pathId uint32, pathAttrs []bgp.PathAttributeInterface, isAddPath bool) *RibEntry { return &RibEntry{ PeerIndex: index, OriginatedTime: time, - PathIdentifier: pathid, - PathAttributes: pathattrs, - isAddPath: pathid != 0, + PathIdentifier: pathId, + PathAttributes: pathAttrs, + isAddPath: isAddPath, } } diff --git a/packet/mrt/mrt_test.go b/packet/mrt/mrt_test.go index e2e601a5..30edd496 100644 --- a/packet/mrt/mrt_test.go +++ b/packet/mrt/mrt_test.go @@ -114,7 +114,7 @@ func TestMrtRibEntry(t *testing.T) { bgp.NewPathAttributeLocalPref(1 << 22), } - e1 := NewRibEntry(1, uint32(time.Now().Unix()), 0, p) + e1 := NewRibEntry(1, uint32(time.Now().Unix()), 0, p, false) b1, err := e1.Serialize() if err != nil { t.Fatal(err) @@ -143,7 +143,7 @@ func TestMrtRibEntryWithAddPath(t *testing.T) { bgp.NewPathAttributeMultiExitDisc(1 << 20), bgp.NewPathAttributeLocalPref(1 << 22), } - e1 := NewRibEntry(1, uint32(time.Now().Unix()), 200, p) + e1 := NewRibEntry(1, uint32(time.Now().Unix()), 200, p, true) b1, err := e1.Serialize() if err != nil { t.Fatal(err) @@ -173,9 +173,9 @@ func TestMrtRib(t *testing.T) { bgp.NewPathAttributeLocalPref(1 << 22), } - e1 := NewRibEntry(1, uint32(time.Now().Unix()), 0, p) - e2 := NewRibEntry(2, uint32(time.Now().Unix()), 0, p) - e3 := NewRibEntry(3, uint32(time.Now().Unix()), 0, p) + e1 := NewRibEntry(1, uint32(time.Now().Unix()), 0, p, false) + e2 := NewRibEntry(2, uint32(time.Now().Unix()), 0, p, false) + e3 := NewRibEntry(3, uint32(time.Now().Unix()), 0, p, false) r1 := NewRib(1, bgp.NewIPAddrPrefix(24, "192.168.0.0"), []*RibEntry{e1, e2, e3}) b1, err := r1.Serialize() @@ -207,9 +207,9 @@ func TestMrtRibWithAddPath(t *testing.T) { bgp.NewPathAttributeLocalPref(1 << 22), } - e1 := NewRibEntry(1, uint32(time.Now().Unix()), 100, p) - e2 := NewRibEntry(2, uint32(time.Now().Unix()), 200, p) - e3 := NewRibEntry(3, uint32(time.Now().Unix()), 300, p) + e1 := NewRibEntry(1, uint32(time.Now().Unix()), 100, p, true) + e2 := NewRibEntry(2, uint32(time.Now().Unix()), 200, p, true) + e3 := NewRibEntry(3, uint32(time.Now().Unix()), 300, p, true) r1 := NewRib(1, bgp.NewIPAddrPrefix(24, "192.168.0.0"), []*RibEntry{e1, e2, e3}) b1, err := r1.Serialize() |