Fix bgp-segment-routing
[bgpcep.git] / bgp / extensions / linkstate / src / main / java / org / opendaylight / protocol / bgp / linkstate / impl / attribute / sr / RangeTlvParser.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.protocol.bgp.linkstate.impl.attribute.sr;
9
10 import static org.opendaylight.protocol.bgp.linkstate.impl.attribute.PrefixAttributesParser.BINDING_SID;
11 import static org.opendaylight.protocol.bgp.linkstate.impl.attribute.sr.binding.sid.sub.tlvs.Ipv4PrefixSidParser.PREFIX_SID;
12 import static org.opendaylight.protocol.bgp.linkstate.impl.attribute.sr.binding.sid.sub.tlvs.Ipv6PrefixSidParser.IPV6_PREFIX_SID;
13 import static org.opendaylight.protocol.bgp.linkstate.impl.attribute.sr.binding.sid.sub.tlvs.SIDParser.SID_TYPE;
14
15 import io.netty.buffer.ByteBuf;
16 import io.netty.buffer.Unpooled;
17 import java.util.ArrayList;
18 import java.util.List;
19 import org.opendaylight.protocol.bgp.linkstate.impl.attribute.sr.SidLabelIndexParser.Size;
20 import org.opendaylight.protocol.bgp.linkstate.spi.TlvUtil;
21 import org.opendaylight.protocol.util.BitArray;
22 import org.opendaylight.protocol.util.ByteBufUtils;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.ProtocolId;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.prefix.state.SrRange;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.prefix.state.SrRangeBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.RangeSubTlv;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.BindingSidTlvCase;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.BindingSidTlvCaseBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.Ipv6PrefixSidTlvCase;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.Ipv6PrefixSidTlvCaseBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.PrefixSidTlvCase;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.PrefixSidTlvCaseBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.SidLabelTlvCase;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.SidLabelTlvCaseBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.ipv6.prefix.sid.tlv._case.Ipv6PrefixSidTlv;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.ipv6.prefix.sid.tlv._case.Ipv6PrefixSidTlvBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.prefix.sid.tlv._case.PrefixSidTlv;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.prefix.sid.tlv._case.PrefixSidTlvBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.tlv.SubTlvs;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.tlv.SubTlvsBuilder;
41 import org.slf4j.Logger;
42 import org.slf4j.LoggerFactory;
43
44 public final class RangeTlvParser {
45     private static final Logger LOG = LoggerFactory.getLogger(RangeTlvParser.class);
46     /* Flags */
47     private static final int FLAGS_SIZE = 8;
48     private static final int INNER_AREA = 0;
49     private static final int RESERVED = 1;
50
51     private RangeTlvParser() {
52
53     }
54
55     public static SrRange parseSrRange(final ByteBuf buffer, final ProtocolId protocolId) {
56         final BitArray flags = BitArray.valueOf(buffer, FLAGS_SIZE);
57         final SrRangeBuilder range = new SrRangeBuilder();
58         if (protocolId.equals(ProtocolId.Ospf)) {
59             range.setInterArea(flags.get(INNER_AREA));
60         } else {
61             range.setInterArea(Boolean.FALSE);
62         }
63         buffer.skipBytes(RESERVED);
64         range.setRangeSize(ByteBufUtils.readUint16(buffer));
65         range.setSubTlvs(parseRangeSubTlvs(buffer, protocolId));
66         return range.build();
67     }
68
69     private static List<SubTlvs> parseRangeSubTlvs(final ByteBuf buffer, final ProtocolId protocolId) {
70         final List<SubTlvs> subTlvs = new ArrayList<>();
71         while (buffer.isReadable()) {
72             final RangeSubTlv subTlvCase;
73             final int type = buffer.readUnsignedShort();
74             final int length = buffer.readUnsignedShort();
75             switch (type) {
76                 case PREFIX_SID:
77                     subTlvCase = new PrefixSidTlvCaseBuilder().setPrefixSidTlv(new PrefixSidTlvBuilder(
78                         SrPrefixAttributesParser.parseSrPrefix(buffer.readSlice(length), protocolId)).build()).build();
79                     break;
80                 case IPV6_PREFIX_SID:
81                     subTlvCase = new Ipv6PrefixSidTlvCaseBuilder().setIpv6PrefixSidTlv(new Ipv6PrefixSidTlvBuilder(
82                         Ipv6SrPrefixAttributesParser.parseSrIpv6Prefix(buffer.readSlice(length))).build()).build();
83                     break;
84                 case BINDING_SID:
85                     subTlvCase = new BindingSidTlvCaseBuilder(
86                         BindingSidLabelParser.parseBindingSidLabel(buffer.readSlice(length), protocolId)).build();
87                     break;
88                 case SID_TYPE:
89                     subTlvCase = new SidLabelTlvCaseBuilder().setSidLabelIndex(
90                         SidLabelIndexParser.parseSidLabelIndex(Size.forValue(length), buffer.readSlice(length)))
91                         .build();
92                     break;
93                 default:
94                     LOG.info("Unknown type of range sub-tlv: {}", type);
95                     buffer.skipBytes(length);
96                     continue;
97             }
98             subTlvs.add(new SubTlvsBuilder().setRangeSubTlv(subTlvCase).build());
99         }
100         return subTlvs;
101     }
102
103     public static void serializeSrRange(final SrRange srRange, final ByteBuf aggregator) {
104         final BitArray flags = new BitArray(FLAGS_SIZE);
105         flags.set(INNER_AREA, srRange.isInterArea());
106         flags.toByteBuf(aggregator);
107         aggregator.writeZero(RESERVED);
108         aggregator.writeShort(srRange.getRangeSize().toJava());
109         serializeSubTlvs(aggregator, srRange.getSubTlvs());
110     }
111
112     private static void serializeSubTlvs(final ByteBuf aggregator, final List<SubTlvs> subTlvs) {
113         for (final SubTlvs subTlv : subTlvs) {
114             ByteBuf buffer = Unpooled.buffer();
115             final RangeSubTlv rangeSubTlv = subTlv.getRangeSubTlv();
116             if (rangeSubTlv instanceof PrefixSidTlvCase) {
117                 final PrefixSidTlv prefixSidTlv = ((PrefixSidTlvCase) rangeSubTlv).getPrefixSidTlv();
118                 SrPrefixAttributesParser.serializePrefixAttributes(
119                     prefixSidTlv.getFlags(),
120                     prefixSidTlv.getAlgorithm(),
121                     prefixSidTlv.getSidLabelIndex(),
122                     buffer);
123                 TlvUtil.writeTLV(PREFIX_SID, buffer, aggregator);
124             } else if (rangeSubTlv instanceof Ipv6PrefixSidTlvCase) {
125                 final Ipv6PrefixSidTlv prefixSidTlv = ((Ipv6PrefixSidTlvCase) rangeSubTlv).getIpv6PrefixSidTlv();
126                 Ipv6SrPrefixAttributesParser.serializePrefixAttributes(prefixSidTlv.getAlgorithm(), buffer);
127                 TlvUtil.writeTLV(IPV6_PREFIX_SID, buffer, aggregator);
128             } else if (rangeSubTlv instanceof BindingSidTlvCase) {
129                 final BindingSidTlvCase rangeTlv = (BindingSidTlvCase) rangeSubTlv;
130                 BindingSidLabelParser.serializeBindingSidAttributes(
131                     rangeTlv.getWeight(),
132                     rangeTlv.getFlags(),
133                     rangeTlv.getBindingSubTlvs(),
134                     buffer);
135                 TlvUtil.writeTLV(BINDING_SID, buffer, aggregator);
136             } else if (rangeSubTlv instanceof SidLabelTlvCase) {
137                 buffer = SidLabelIndexParser.serializeSidValue(((SidLabelTlvCase) rangeSubTlv).getSidLabelIndex());
138                 TlvUtil.writeTLV(SID_TYPE, buffer, aggregator);
139             }
140         }
141     }
142
143 }