BUG-4552: Enhance LS-SR - discern protocols
[bgpcep.git] / bgp / linkstate / src / main / java / org / opendaylight / protocol / bgp / linkstate / 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.attribute.sr;
9
10 import io.netty.buffer.ByteBuf;
11 import io.netty.buffer.Unpooled;
12 import java.util.ArrayList;
13 import java.util.List;
14 import org.opendaylight.protocol.bgp.linkstate.attribute.PrefixAttributesParser;
15 import org.opendaylight.protocol.bgp.linkstate.attribute.sr.SidLabelIndexParser.Size;
16 import org.opendaylight.protocol.bgp.linkstate.spi.TlvUtil;
17 import org.opendaylight.protocol.util.BitArray;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.ProtocolId;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.prefix.state.SrRange;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.prefix.state.SrRangeBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.RangeSubTlv;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.BindingSidTlvCase;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.BindingSidTlvCaseBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.PrefixSidTlvCase;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.PrefixSidTlvCaseBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.SidLabelTlvCase;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.sub.tlvs.range.sub.tlv.SidLabelTlvCaseBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.tlv.SubTlvs;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.segment.routing.ext.rev151014.range.tlv.SubTlvsBuilder;
30 import org.slf4j.Logger;
31 import org.slf4j.LoggerFactory;
32
33 public final class RangeTlvParser {
34
35     private static final Logger LOG = LoggerFactory.getLogger(RangeTlvParser.class);
36
37     private RangeTlvParser() {
38         throw new UnsupportedOperationException();
39     }
40
41     /* Flags */
42     private static final int FLAGS_SIZE = 8;
43     private static final int INNER_AREA = 0;
44
45     private static final int RESERVED = 1;
46
47     public static SrRange parseSrRange(final ByteBuf buffer, final ProtocolId protocolId) {
48         final BitArray flags = BitArray.valueOf(buffer, FLAGS_SIZE);
49         final SrRangeBuilder range = new SrRangeBuilder();
50         range.setInterArea(flags.get(INNER_AREA));
51         buffer.skipBytes(RESERVED);
52         range.setRangeSize(buffer.readUnsignedShort());
53         range.setSubTlvs(parseRangeSubTlvs(buffer, protocolId));
54         return range.build();
55     }
56
57     private static List<SubTlvs> parseRangeSubTlvs(final ByteBuf buffer, final ProtocolId protocolId) {
58         final List<SubTlvs> subTlvs = new ArrayList<>();
59         while (buffer.isReadable()) {
60             final SubTlvsBuilder subTlv = new SubTlvsBuilder();
61             RangeSubTlv subTlvCase = null;
62             final int type = buffer.readUnsignedShort();
63             final int length = buffer.readUnsignedShort();
64             switch (type) {
65             case PrefixAttributesParser.PREFIX_SID:
66                 subTlvCase = new PrefixSidTlvCaseBuilder(SrPrefixAttributesParser.parseSrPrefix(buffer.readSlice(length), protocolId)).build();
67                 break;
68             case PrefixAttributesParser.BINDING_SID:
69                 subTlvCase = new BindingSidTlvCaseBuilder(BindingSidLabelParser.parseBindingSidLabel(buffer.readSlice(length), protocolId)).build();
70                 break;
71             case SidLabelIndexParser.SID_TYPE:
72                 subTlvCase = new SidLabelTlvCaseBuilder().setSidLabelIndex(SidLabelIndexParser.parseSidLabelIndex(Size.forValue(length), buffer.readSlice(length))).build();
73                 break;
74             default:
75                 LOG.info("Unknown type of range sub-tlv: {}", type);
76                 buffer.skipBytes(length);
77                 continue;
78             }
79             subTlvs.add(subTlv.setRangeSubTlv(subTlvCase).build());
80         }
81         return subTlvs;
82     }
83
84     public static void serializeSrRange(final SrRange srRange, final ByteBuf aggregator) {
85         final BitArray flags = new BitArray(FLAGS_SIZE);
86         flags.set(INNER_AREA, srRange.isInterArea());
87         flags.toByteBuf(aggregator);
88         aggregator.writeZero(RESERVED);
89         aggregator.writeShort(srRange.getRangeSize());
90         serializeSubTlvs(aggregator, srRange.getSubTlvs());
91     }
92
93     private static void serializeSubTlvs(final ByteBuf aggregator, final List<SubTlvs> subTlvs) {
94         for (final SubTlvs subTlv : subTlvs) {
95             ByteBuf buffer = Unpooled.buffer();
96             final RangeSubTlv rangeSubTlv = subTlv.getRangeSubTlv();
97             if (rangeSubTlv instanceof PrefixSidTlvCase) {
98                 final PrefixSidTlvCase prefixSidTlv = (PrefixSidTlvCase) rangeSubTlv;
99                 SrPrefixAttributesParser.serializePrefixAttributes(
100                     prefixSidTlv.getFlags(),
101                     prefixSidTlv.getAlgorithm(),
102                     prefixSidTlv.getSidLabelIndex(),
103                     buffer);
104                 TlvUtil.writeTLV(PrefixAttributesParser.PREFIX_SID, buffer, aggregator);
105             } else if (rangeSubTlv instanceof BindingSidTlvCase) {
106                 final BindingSidTlvCase rangeTlv = (BindingSidTlvCase) rangeSubTlv;
107                 BindingSidLabelParser.serializeBindingSidAttributes(
108                     rangeTlv.getWeight(),
109                     rangeTlv.getFlags(),
110                     rangeTlv.getBindingSubTlvs(),
111                     buffer);
112                 TlvUtil.writeTLV(PrefixAttributesParser.BINDING_SID, buffer, aggregator);
113             } else if (rangeSubTlv instanceof SidLabelTlvCase) {
114                 buffer = SidLabelIndexParser.serializeSidValue(((SidLabelTlvCase) rangeSubTlv).getSidLabelIndex());
115                 TlvUtil.writeTLV(SidLabelIndexParser.SID_TYPE, buffer, aggregator);
116             }
117         }
118     }
119
120 }