Newer
Older
bnet "github.com/bio-routing/bio-rd/net"
"github.com/bio-routing/bio-rd/protocols/bgp/types"
"github.com/bio-routing/bio-rd/route"
func decodePathAttrs(buf *bytes.Buffer, tpal uint16, opt *types.Options) (*PathAttribute, error) {
var ret *PathAttribute
var eol *PathAttribute
var pa *PathAttribute
var err error
var consumed uint16
p := uint16(0)
for p < tpal {
pa, consumed, err = decodePathAttr(buf, opt)
if err != nil {
return nil, fmt.Errorf("Unable to decode path attr: %v", err)
}
p += consumed
if ret == nil {
ret = pa
eol = pa
} else {
eol.Next = pa
eol = pa
}
}
return ret, nil
}
func decodePathAttr(buf *bytes.Buffer, opt *types.Options) (pa *PathAttribute, consumed uint16, err error) {
pa = &PathAttribute{}
err = decodePathAttrFlags(buf, pa)
if err != nil {
return nil, consumed, fmt.Errorf("Unable to get path attribute flags: %v", err)
}
consumed++
err = decode(buf, []interface{}{&pa.TypeCode})
if err != nil {
return nil, consumed, err
}
consumed++
n, err := pa.setLength(buf)
if err != nil {
return nil, consumed, err
}
consumed += uint16(n)
switch pa.TypeCode {
case OriginAttr:
if err := pa.decodeOrigin(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode Origin: %v", err)
}
case ASPathAttr:
asnLength := uint8(2)
if opt.Supports4OctetASN {
asnLength = 4
}
if err := pa.decodeASPath(buf, asnLength); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode AS Path: %v", err)
}
/* Don't decodeAS4Paths yet: The rest of the software does not support it right yet!
case AS4PathAttr:
if err := pa.decodeASPath(buf, 4); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode AS4 Path: %v", err)
case NextHopAttr:
if err := pa.decodeNextHop(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode Next-Hop: %v", err)
}
case MEDAttr:
if err := pa.decodeMED(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode MED: %v", err)
}
case LocalPrefAttr:
if err := pa.decodeLocalPref(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode local pref: %v", err)
}
case AggregatorAttr:
if err := pa.decodeAggregator(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode Aggregator: %v", err)
}
case AtomicAggrAttr:
// Nothing to do for 0 octet long attribute
case CommunitiesAttr:
if err := pa.decodeCommunities(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode Community: %v", err)
}
case OriginatorIDAttr:
if err := pa.decodeOriginatorID(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode OriginatorID: %v", err)
}
case ClusterListAttr:
if err := pa.decodeClusterList(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode OriginatorID: %v", err)
}
case MultiProtocolReachNLRICode:
if err := pa.decodeMultiProtocolReachNLRI(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to multi protocol reachable NLRI: %v", err)
}
case MultiProtocolUnreachNLRICode:
if err := pa.decodeMultiProtocolUnreachNLRI(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to multi protocol unreachable NLRI: %v", err)
}
case AS4AggregatorAttr:
if err := pa.decodeAS4Aggregator(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to skip not supported AS4Aggregator: %v", err)
}
if err := pa.decodeLargeCommunities(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode large communities: %v", err)
}
if err := pa.decodeUnknown(buf); err != nil {
return nil, consumed, fmt.Errorf("Failed to decode unknown attribute: %v", err)
}
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
func (pa *PathAttribute) decodeMultiProtocolReachNLRI(buf *bytes.Buffer) error {
b := make([]byte, pa.Length)
n, err := buf.Read(b)
if err != nil {
return fmt.Errorf("Unable to read %d bytes from buffer: %v", pa.Length, err)
}
if n != int(pa.Length) {
return fmt.Errorf("Unable to read %d bytes from buffer, only got %d bytes", pa.Length, n)
}
nlri, err := deserializeMultiProtocolReachNLRI(b)
if err != nil {
return fmt.Errorf("Unable to decode MP_REACH_NLRI: %v", err)
}
pa.Value = nlri
return nil
}
func (pa *PathAttribute) decodeMultiProtocolUnreachNLRI(buf *bytes.Buffer) error {
b := make([]byte, pa.Length)
n, err := buf.Read(b)
if err != nil {
return fmt.Errorf("Unable to read %d bytes from buffer: %v", pa.Length, err)
}
if n != int(pa.Length) {
return fmt.Errorf("Unable to read %d bytes from buffer, only got %d bytes", pa.Length, n)
}
nlri, err := deserializeMultiProtocolUnreachNLRI(b)
if err != nil {
return fmt.Errorf("Unable to decode MP_UNREACH_NLRI: %v", err)
}
pa.Value = nlri
return nil
}
func (pa *PathAttribute) decodeUnknown(buf *bytes.Buffer) error {
u := make([]byte, pa.Length)
err := decode(buf, []interface{}{&u})
if err != nil {
return fmt.Errorf("Unable to decode: %v", err)
}
pa.Value = u
return nil
}
func (pa *PathAttribute) decodeOrigin(buf *bytes.Buffer) error {
origin := uint8(0)
p := uint16(0)
err := decode(buf, []interface{}{&origin})
if err != nil {
return fmt.Errorf("Unable to decode: %v", err)
}
pa.Value = origin
p++
return dumpNBytes(buf, pa.Length-p)
}
func (pa *PathAttribute) decodeASPath(buf *bytes.Buffer, asnLength uint8) error {
if segment.Type != types.ASSet && segment.Type != types.ASSequence {
return fmt.Errorf("Invalid AS Path segment type: %d", segment.Type)
}
if count == 0 {
return fmt.Errorf("Invalid AS Path segment length: %d", count)
segment.ASNs = make([]uint32, count)
for i := uint8(0); i < count; i++ {
asn, err := pa.decodeASN(buf, asnLength)
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
func (pa *PathAttribute) decodeASN(buf *bytes.Buffer, asnSize uint8) (asn uint32, err error) {
if asnSize == 4 {
return pa.decode4ByteASN(buf)
}
return pa.decode2ByteASN(buf)
}
func (pa *PathAttribute) decode4ByteASN(buf *bytes.Buffer) (asn uint32, err error) {
asn4 := uint32(0)
err = decode(buf, []interface{}{&asn4})
if err != nil {
return 0, err
}
return uint32(asn4), nil
}
func (pa *PathAttribute) decode2ByteASN(buf *bytes.Buffer) (asn uint32, err error) {
asn4 := uint16(0)
err = decode(buf, []interface{}{&asn4})
if err != nil {
return 0, err
}
return uint32(asn4), nil
}
nextHop := uint32(0)
err := decode(buf, []interface{}{&nextHop})
if err != nil {
return fmt.Errorf("Unable to decode next hop: %v", err)
}
pa.Value = bnet.IPv4(nextHop)
return nil
}
func (pa *PathAttribute) decodeMED(buf *bytes.Buffer) error {
return pa.decodeUint32(buf, "MED")
}
func (pa *PathAttribute) decodeLocalPref(buf *bytes.Buffer) error {
return pa.decodeUint32(buf, "local pref")
}
func (pa *PathAttribute) decodeAggregator(buf *bytes.Buffer) error {
err := decode(buf, []interface{}{&aggr.ASN, &aggr.Address})
func (pa *PathAttribute) decodeCommunities(buf *bytes.Buffer) error {
return fmt.Errorf("Unable to read community path attribute. Length %d is not divisible by 4", pa.Length)
count := pa.Length / CommunityLen
coms := make([]uint32, count)
for i := uint16(0); i < count; i++ {
v, err := read4BytesAsUint32(buf)
func (pa *PathAttribute) decodeLargeCommunities(buf *bytes.Buffer) error {
if pa.Length%LargeCommunityLen != 0 {
return fmt.Errorf("Unable to read large community path attribute. Length %d is not divisible by 12", pa.Length)
}
count := pa.Length / LargeCommunityLen
v, err := read4BytesAsUint32(buf)
if err != nil {
return err
}
com.GlobalAdministrator = v
if err != nil {
return err
}
com.DataPart1 = v
if err != nil {
return err
}
com.DataPart2 = v
coms[i] = com
}
pa.Value = coms
func (pa *PathAttribute) decodeAS4Aggregator(buf *bytes.Buffer) error {
return pa.decodeUint32(buf, "AS4Aggregator")
}
func (pa *PathAttribute) decodeUint32(buf *bytes.Buffer, attrName string) error {
v, err := read4BytesAsUint32(buf)
if err != nil {
return fmt.Errorf("Unable to decode %s: %v", attrName, err)
p := uint16(4)
err = dumpNBytes(buf, pa.Length-p)
if err != nil {
return fmt.Errorf("dumpNBytes failed: %v", err)
}
return nil
}
func (pa *PathAttribute) decodeOriginatorID(buf *bytes.Buffer) error {
return pa.decodeUint32(buf, "OriginatorID")
}
func (pa *PathAttribute) decodeClusterList(buf *bytes.Buffer) error {
if pa.Length%ClusterIDLen != 0 {
return fmt.Errorf("Unable to read ClusterList path attribute. Length %d is not divisible by %d", pa.Length, ClusterIDLen)
}
count := pa.Length / ClusterIDLen
cids := make([]uint32, count)
for i := uint16(0); i < count; i++ {
v, err := read4BytesAsUint32(buf)
if err != nil {
return err
}
cids[i] = v
}
pa.Value = cids
return nil
}
func (pa *PathAttribute) setLength(buf *bytes.Buffer) (int, error) {
bytesRead := 0
if pa.ExtendedLength {
err := decode(buf, []interface{}{&pa.Length})
if err != nil {
return 0, err
}
bytesRead = 2
} else {
x := uint8(0)
err := decode(buf, []interface{}{&x})
if err != nil {
return 0, err
}
pa.Length = uint16(x)
bytesRead = 1
}
return bytesRead, nil
}
func (pa *PathAttribute) Copy() *PathAttribute {
return &PathAttribute{
ExtendedLength: pa.ExtendedLength,
Length: pa.Length,
Optional: pa.Optional,
Partial: pa.Partial,
Transitive: pa.Transitive,
TypeCode: pa.TypeCode,
Value: pa.Value,
}
}
// dumpNBytes is used to dump n bytes of buf. This is useful in case an path attributes
// length doesn't match a fixed length's attributes length (e.g. ORIGIN is always an octet)
func dumpNBytes(buf *bytes.Buffer, n uint16) error {
if n <= 0 {
return nil
}
dump := make([]byte, n)
err := decode(buf, []interface{}{&dump})
if err != nil {
return err
}
return nil
}
Daniel Czerwonk
committed
func (pa *PathAttribute) Serialize(buf *bytes.Buffer, opt *types.Options) uint16 {
pathAttrLen := uint16(0)
switch pa.TypeCode {
case OriginAttr:
pathAttrLen = uint16(pa.serializeOrigin(buf))
pathAttrLen = uint16(pa.serializeASPath(buf, opt))
pathAttrLen = uint16(pa.serializeNextHop(buf))
pathAttrLen = uint16(pa.serializeMED(buf))
pathAttrLen = uint16(pa.serializeLocalpref(buf))
pathAttrLen = uint16(pa.serializeAtomicAggregate(buf))
pathAttrLen = uint16(pa.serializeAggregator(buf))
pathAttrLen = uint16(pa.serializeCommunities(buf))
pathAttrLen = uint16(pa.serializeLargeCommunities(buf))
Maximilian Wilhelm
committed
case OriginatorIDAttr:
pathAttrLen = uint16(pa.serializeOriginatorID(buf))
case ClusterListAttr:
pathAttrLen = uint16(pa.serializeClusterList(buf))
default:
pathAttrLen = pa.serializeUnknownAttribute(buf)
}
return pathAttrLen
}
func (pa *PathAttribute) serializeOrigin(buf *bytes.Buffer) uint8 {
attrFlags := uint8(0)
attrFlags = setTransitive(attrFlags)
buf.WriteByte(attrFlags)
buf.WriteByte(OriginAttr)
length := uint8(1)
buf.WriteByte(length)
buf.WriteByte(pa.Value.(uint8))
return 4
}
func (pa *PathAttribute) serializeASPath(buf *bytes.Buffer, opt *types.Options) uint8 {
attrFlags := uint8(0)
attrFlags = setTransitive(attrFlags)
buf.WriteByte(attrFlags)
buf.WriteByte(ASPathAttr)
asnLength := uint8(2)
if opt.Supports4OctetASN {
asnLength = 4
}
length := uint8(0)
segmentsBuf := bytes.NewBuffer(nil)
segmentsBuf.WriteByte(segment.Type)
segmentsBuf.WriteByte(uint8(len(segment.ASNs)))
if asnLength == 2 {
segmentsBuf.Write(convert.Uint16Byte(uint16(asn)))
} else {
segmentsBuf.Write(convert.Uint32Byte(asn))
}
length += 2 + uint8(len(segment.ASNs))*asnLength
buf.WriteByte(length)
buf.Write(segmentsBuf.Bytes())
return length + 2
}
func (pa *PathAttribute) serializeNextHop(buf *bytes.Buffer) uint8 {
attrFlags := uint8(0)
attrFlags = setTransitive(attrFlags)
buf.WriteByte(attrFlags)
buf.WriteByte(NextHopAttr)
length := uint8(4)
buf.WriteByte(length)
addr := pa.Value.(bnet.IP)
buf.Write(addr.Bytes())
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
return 7
}
func (pa *PathAttribute) serializeMED(buf *bytes.Buffer) uint8 {
attrFlags := uint8(0)
attrFlags = setOptional(attrFlags)
buf.WriteByte(attrFlags)
buf.WriteByte(MEDAttr)
length := uint8(4)
buf.WriteByte(length)
buf.Write(convert.Uint32Byte(pa.Value.(uint32)))
return 7
}
func (pa *PathAttribute) serializeLocalpref(buf *bytes.Buffer) uint8 {
attrFlags := uint8(0)
attrFlags = setTransitive(attrFlags)
buf.WriteByte(attrFlags)
buf.WriteByte(LocalPrefAttr)
length := uint8(4)
buf.WriteByte(length)
buf.Write(convert.Uint32Byte(pa.Value.(uint32)))
return 7
}
func (pa *PathAttribute) serializeAtomicAggregate(buf *bytes.Buffer) uint8 {
attrFlags := uint8(0)
attrFlags = setTransitive(attrFlags)
buf.WriteByte(attrFlags)
buf.WriteByte(AtomicAggrAttr)
length := uint8(0)
buf.WriteByte(length)
return 3
}
func (pa *PathAttribute) serializeAggregator(buf *bytes.Buffer) uint8 {
attrFlags := uint8(0)
attrFlags = setOptional(attrFlags)
attrFlags = setTransitive(attrFlags)
buf.WriteByte(attrFlags)
buf.WriteByte(AggregatorAttr)
aggregator := pa.Value.(types.Aggregator)
buf.Write(convert.Uint16Byte(aggregator.ASN))
buf.Write(convert.Uint32Byte(aggregator.Address))
return 9
func (pa *PathAttribute) serializeCommunities(buf *bytes.Buffer) uint8 {
coms := pa.Value.([]uint32)
if len(coms) == 0 {
return 0
}
attrFlags := uint8(0)
attrFlags = setOptional(attrFlags)
attrFlags = setTransitive(attrFlags)
attrFlags = setPartial(attrFlags)
buf.WriteByte(attrFlags)
buf.WriteByte(CommunitiesAttr)
length := uint8(CommunityLen * len(coms))
buf.WriteByte(length)
for _, com := range coms {
buf.Write(convert.Uint32Byte(com))
}
return length
}
func (pa *PathAttribute) serializeLargeCommunities(buf *bytes.Buffer) uint8 {
if len(coms) == 0 {
return 0
}
attrFlags := uint8(0)
attrFlags = setOptional(attrFlags)
attrFlags = setTransitive(attrFlags)
attrFlags = setPartial(attrFlags)
buf.WriteByte(attrFlags)
length := uint8(LargeCommunityLen * len(coms))
buf.WriteByte(length)
for _, com := range coms {
buf.Write(convert.Uint32Byte(com.GlobalAdministrator))
buf.Write(convert.Uint32Byte(com.DataPart1))
buf.Write(convert.Uint32Byte(com.DataPart2))
}
return length
}
Maximilian Wilhelm
committed
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
func (pa *PathAttribute) serializeOriginatorID(buf *bytes.Buffer) uint8 {
attrFlags := uint8(0)
attrFlags = setOptional(attrFlags)
buf.WriteByte(attrFlags)
buf.WriteByte(OriginatorIDAttr)
length := uint8(4)
buf.WriteByte(length)
oid := pa.Value.(uint32)
buf.Write(convert.Uint32Byte(oid))
return 7
}
func (pa *PathAttribute) serializeClusterList(buf *bytes.Buffer) uint8 {
cids := pa.Value.([]uint32)
if len(cids) == 0 {
return 0
}
attrFlags := uint8(0)
attrFlags = setOptional(attrFlags)
buf.WriteByte(attrFlags)
buf.WriteByte(ClusterListAttr)
length := uint8(ClusterIDLen * len(cids))
buf.WriteByte(length)
for _, cid := range cids {
buf.Write(convert.Uint32Byte(cid))
}
return length
}
func (pa *PathAttribute) serializeUnknownAttribute(buf *bytes.Buffer) uint16 {
attrFlags := uint8(0)
if pa.Optional {
attrFlags = setOptional(attrFlags)
}
if pa.ExtendedLength {
attrFlags = setExtendedLength(attrFlags)
}
attrFlags = setTransitive(attrFlags)
buf.WriteByte(attrFlags)
buf.WriteByte(pa.TypeCode)
b := pa.Value.([]byte)
if pa.ExtendedLength {
l := len(b)
buf.WriteByte(uint8(l >> 8))
buf.WriteByte(uint8(l & 0x0000FFFF))
} else {
buf.WriteByte(uint8(len(b)))
}
return uint16(len(b) + 2)
func fourBytesToUint32(address [4]byte) uint32 {
return uint32(address[0])<<24 + uint32(address[1])<<16 + uint32(address[2])<<8 + uint32(address[3])
}
func read4BytesAsUint32(buf *bytes.Buffer) (uint32, error) {
b := [4]byte{}
n, err := buf.Read(b[:])
if err != nil {
return 0, err
}
if n != 4 {
return 0, fmt.Errorf("Unable to read as uint32. Expected 4 bytes but got only %d", n)
}
return fourBytesToUint32(b), nil
}
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
// AddOptionalPathAttributes adds optional path attributes to linked list pa
func (pa *PathAttribute) AddOptionalPathAttributes(p *route.Path) *PathAttribute {
current := pa
if len(p.BGPPath.Communities) > 0 {
communities := &PathAttribute{
TypeCode: CommunitiesAttr,
Value: p.BGPPath.Communities,
}
current.Next = communities
current = communities
}
if len(p.BGPPath.LargeCommunities) > 0 {
largeCommunities := &PathAttribute{
TypeCode: LargeCommunitiesAttr,
Value: p.BGPPath.LargeCommunities,
}
current.Next = largeCommunities
current = largeCommunities
}
return current
}
// PathAttributes converts a path object into a linked list of path attributes
Maximilian Wilhelm
committed
func PathAttributes(p *route.Path, iBGP bool, rrClient bool) (*PathAttribute, error) {
asPath := &PathAttribute{
TypeCode: ASPathAttr,
Value: p.BGPPath.ASPath,
}
origin := &PathAttribute{
TypeCode: OriginAttr,
Value: p.BGPPath.Origin,
}
last.Next = origin
last = origin
nextHop := &PathAttribute{
TypeCode: NextHopAttr,
Value: p.BGPPath.NextHop,
}
last.Next = nextHop
last = nextHop
if p.BGPPath.AtomicAggregate {
atomicAggr := &PathAttribute{
TypeCode: AtomicAggrAttr,
}
last.Next = atomicAggr
last = atomicAggr
}
if p.BGPPath.Aggregator != nil {
aggregator := &PathAttribute{
TypeCode: AggregatorAttr,
Value: p.BGPPath.Aggregator,
}
last.Next = aggregator
last = aggregator
}
if iBGP {
localPref := &PathAttribute{
TypeCode: LocalPrefAttr,
Value: p.BGPPath.LocalPref,
}
last.Next = localPref
last = localPref
Maximilian Wilhelm
committed
if rrClient {
originatorID := &PathAttribute{
TypeCode: OriginatorIDAttr,
Value: p.BGPPath.OriginatorID,
}
last.Next = originatorID
last = originatorID
clusterList := &PathAttribute{
TypeCode: ClusterListAttr,
Value: p.BGPPath.ClusterList,
}
last.Next = clusterList
last = clusterList
}
optionals := last.AddOptionalPathAttributes(p)
for _, unknownAttr := range p.BGPPath.UnknownAttributes {
last.Next = &PathAttribute{
TypeCode: unknownAttr.TypeCode,
Optional: unknownAttr.Optional,
Transitive: unknownAttr.Transitive,
Partial: unknownAttr.Partial,
Value: unknownAttr.Value,
}
last = last.Next
}
return asPath, nil
}