mirror/osrg-gobgp
1
0
Fork 0
mirror of https://github.com/osrg/gobgp.git synced 2024-05-11 05:55:10 +00:00
Code Issues Releases Activity

Bmp changes

Browse Source
This commit is contained in:
mgaitonde
2022-04-28 20:31:07 -07:00
committed by FUJITA Tomonori
parent fc6fa7c64e
commit 6d3dfb478c
2 changed files with 48 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/packet/bmp/bmp.go
+29 -13
View File
@@ -51,9 +51,15 @@ const (
BMP_PEER_FLAG_IPV6 = 1 << 7
BMP_PEER_FLAG_POST_POLICY = 1 << 6
BMP_PEER_FLAG_TWO_AS = 1 << 5
BMP_PEER_FLAG_FILTERED = 1 << 6
BMP_PEER_FLAG_ADJ_RIB_TYP = 1 << 4
)
func makeIP(data []byte) net.IP {
ip := make(net.IP, len(data))
copy(ip, data)
return ip
}
func (h *BMPHeader) DecodeFromBytes(data []byte) error {
h.Version = data[0]
if data[0] != BMP_VERSION {
@@ -108,14 +114,18 @@ func (h *BMPPeerHeader) IsPostPolicy() bool {
}
}
func (h *BMPPeerHeader) IsAdjRIBOut() bool {
return h.Flags&BMP_PEER_FLAG_ADJ_RIB_TYP != 0
}
func (h *BMPPeerHeader) DecodeFromBytes(data []byte) error {
h.PeerType = data[0]
h.Flags = data[1]
h.PeerDistinguisher = binary.BigEndian.Uint64(data[2:10])
if h.Flags&BMP_PEER_FLAG_IPV6 != 0 {
h.PeerAddress = net.IP(data[10:26]).To16()
h.PeerAddress = makeIP(data[10:26]).To16()
} else {
h.PeerAddress = net.IP(data[22:26]).To4()
h.PeerAddress = makeIP(data[22:26]).To4()
}
h.PeerAS = binary.BigEndian.Uint32(data[26:30])
h.PeerBGPID = data[30:34]
@@ -193,6 +203,10 @@ const (
BMP_STAT_TYPE_WITHDRAW_UPDATE
BMP_STAT_TYPE_WITHDRAW_PREFIX
BMP_STAT_TYPE_DUPLICATE_UPDATE
BMP_STAT_TYPE_ADJ_RIB_OUT_PRE_POLICY
BMP_STAT_TYPE_ADJ_RIB_OUT_POST_POLICY
BMP_STAT_TYPE_PER_AFI_SAFI_ADJ_RIB_OUT_PRE_POLICY
BMP_STAT_TYPE_PER_AFI_SAFI_ADJ_RIB_OUT_POST_POLICY
)
type BMPStatsTLVInterface interface {
@@ -339,9 +353,11 @@ func (body *BMPStatisticsReport) ParseBody(msg *BMPMessage, data []byte) error {
}
var s BMPStatsTLVInterface
switch tl.Type {
case BMP_STAT_TYPE_ADJ_RIB_IN, BMP_STAT_TYPE_LOC_RIB:
case BMP_STAT_TYPE_ADJ_RIB_IN, BMP_STAT_TYPE_LOC_RIB, BMP_STAT_TYPE_ADJ_RIB_OUT_PRE_POLICY,
BMP_STAT_TYPE_ADJ_RIB_OUT_POST_POLICY:
s = &BMPStatsTLV64{BMPStatsTLV: tl}
case BMP_STAT_TYPE_PER_AFI_SAFI_ADJ_RIB_IN, BMP_STAT_TYPE_PER_AFI_SAFI_LOC_RIB:
case BMP_STAT_TYPE_PER_AFI_SAFI_ADJ_RIB_IN, BMP_STAT_TYPE_PER_AFI_SAFI_LOC_RIB,
BMP_STAT_TYPE_PER_AFI_SAFI_ADJ_RIB_OUT_PRE_POLICY, BMP_STAT_TYPE_PER_AFI_SAFI_ADJ_RIB_OUT_POST_POLICY:
s = &BMPStatsTLVPerAfiSafi64{BMPStatsTLV: tl}
case BMP_STAT_TYPE_REJECTED, BMP_STAT_TYPE_DUPLICATE_PREFIX,
BMP_STAT_TYPE_DUPLICATE_WITHDRAW, BMP_STAT_TYPE_INV_UPDATE_DUE_TO_CLUSTER_LIST_LOOP,
@@ -487,9 +503,9 @@ func NewBMPPeerUpNotification(p BMPPeerHeader, lAddr string, lPort, rPort uint16
func (body *BMPPeerUpNotification) ParseBody(msg *BMPMessage, data []byte) error {
if msg.PeerHeader.Flags&BMP_PEER_FLAG_IPV6 != 0 {
body.LocalAddress = net.IP(data[:16]).To16()
body.LocalAddress = makeIP(data[:16]).To16()
} else {
body.LocalAddress = net.IP(data[12:16]).To4()
body.LocalAddress = makeIP(data[12:16]).To4()
}
body.LocalPort = binary.BigEndian.Uint16(data[16:18])
@@ -1078,12 +1094,12 @@ func SplitBMP(data []byte, atEOF bool) (advance int, token []byte, err error) {
if atEOF && len(data) == 0 || len(data) < BMP_HEADER_SIZE {
return 0, nil, nil
}
msg := &BMPMessage{}
msg.Header.DecodeFromBytes(data)
if uint32(len(data)) < msg.Header.Length {
tmpHdr := &BMPHeader{}
if err = tmpHdr.DecodeFromBytes(data[:BMP_HEADER_SIZE]); err != nil {
return 0, nil, nil
}
return int(msg.Header.Length), data[0:msg.Header.Length], nil
if len(data) < int(tmpHdr.Length) {
return 0, nil, nil
}
return int(tmpHdr.Length), data[0:tmpHdr.Length], nil
}
pkg/packet/bmp/bmp_test.go
+19
View File
@@ -71,6 +71,13 @@ func Test_RouteMonitoring(t *testing.T) {
verify(t, NewBMPRouteMonitoring(*p0, m))
}
func Test_RouteMonitoringAdjRIBOut(t *testing.T) {
m := bgp.NewTestBGPUpdateMessage()
p0 := NewBMPPeerHeader(0, 16, 1000, "10.0.0.1", 12345, "10.0.0.2", 1)
assert.True(t, p0.IsAdjRIBOut())
verify(t, NewBMPRouteMonitoring(*p0, m))
}
func Test_StatisticsReport(t *testing.T) {
p0 := NewBMPPeerHeader(0, 0, 1000, "10.0.0.1", 70000, "10.0.0.2", 1)
s0 := NewBMPStatisticsReport(
@@ -84,6 +91,18 @@ func Test_StatisticsReport(t *testing.T) {
verify(t, s0)
}
func Test_StatisticsReportAdjRIBOut(t *testing.T) {
p0 := NewBMPPeerHeader(0, 8, 1000, "10.0.0.1", 12345, "10.0.0.2", 1)
s0 := NewBMPStatisticsReport(
*p0,
[]BMPStatsTLVInterface{
NewBMPStatsTLV64(BMP_STAT_TYPE_ADJ_RIB_OUT_POST_POLICY, 200),
NewBMPStatsTLVPerAfiSafi64(BMP_STAT_TYPE_PER_AFI_SAFI_ADJ_RIB_OUT_POST_POLICY, bgp.AFI_IP, bgp.SAFI_UNICAST, 300),
},
)
verify(t, s0)
}
func Test_RouteMirroring(t *testing.T) {
p0 := NewBMPPeerHeader(0, 0, 1000, "10.0.0.1", 70000, "10.0.0.2", 1)
s0 := NewBMPRouteMirroring(