1e862816c951136127b76171be8a7388432f62ce
[bgpcep.git] / bgp / extensions / linkstate / src / main / java / org / opendaylight / protocol / bgp / linkstate / impl / nlri / LinkNlriParser.java
1 /*
2  * Copyright (c) 2014 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.nlri;
9
10 import com.google.common.annotations.VisibleForTesting;
11 import io.netty.buffer.ByteBuf;
12 import java.util.Map;
13 import org.opendaylight.protocol.bgp.linkstate.impl.tlvs.Ipv4InterfaceTlvParser;
14 import org.opendaylight.protocol.bgp.linkstate.impl.tlvs.Ipv4NeighborTlvParser;
15 import org.opendaylight.protocol.bgp.linkstate.impl.tlvs.Ipv6InterfaceTlvParser;
16 import org.opendaylight.protocol.bgp.linkstate.impl.tlvs.Ipv6NeighborTlvParser;
17 import org.opendaylight.protocol.bgp.linkstate.impl.tlvs.MultiTopoIdTlvParser;
18 import org.opendaylight.protocol.bgp.linkstate.spi.AbstractNlriTypeCodec;
19 import org.opendaylight.protocol.bgp.linkstate.spi.TlvUtil;
20 import org.opendaylight.protocol.bgp.linkstate.spi.pojo.SimpleNlriTypeRegistry;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.Ipv4InterfaceIdentifier;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.Ipv6InterfaceIdentifier;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.LinkLrIdentifiers;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.NlriType;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.TopologyIdentifier;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.ObjectType;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.LinkCase;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.LinkCaseBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.link._case.LinkDescriptors;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.link._case.LinkDescriptorsBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.link._case.LocalNodeDescriptors;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.link._case.RemoteNodeDescriptors;
33 import org.opendaylight.yangtools.yang.common.QName;
34 import org.opendaylight.yangtools.yang.common.Uint32;
35 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
36 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
37
38 public final class LinkNlriParser extends AbstractNlriTypeCodec {
39
40     /* Link Descriptor QNames */
41     @VisibleForTesting
42     public static final NodeIdentifier IPV4_IFACE_NID = NodeIdentifier.create(
43         Ipv4InterfaceTlvParser.IPV4_IFACE_ADDRESS_QNAME);
44     @VisibleForTesting
45     public static final NodeIdentifier IPV4_NEIGHBOR_NID = NodeIdentifier.create(
46         Ipv4NeighborTlvParser.IPV4_NEIGHBOR_ADDRESS_QNAME);
47     @VisibleForTesting
48     public static final NodeIdentifier LINK_LOCAL_NID = NodeIdentifier.create(
49         QName.create(LinkDescriptors.QNAME, "link-local-identifier").intern());
50     @VisibleForTesting
51     public static final NodeIdentifier LINK_REMOTE_NID = NodeIdentifier.create(
52         QName.create(LinkDescriptors.QNAME, "link-remote-identifier").intern());
53     private static final NodeIdentifier IPV6_IFACE_NID = NodeIdentifier.create(
54         Ipv6InterfaceTlvParser.IPV6_IFACE_ADDRESS_QNAME);
55     private static final NodeIdentifier IPV6_NEIGHBOR_NID = NodeIdentifier.create(
56         Ipv6NeighborTlvParser.IPV6_NEIGHBOR_ADDRESS_QNAME);
57
58     @FunctionalInterface
59     private interface SerializerInterface {
60         void check(Object cont);
61     }
62
63     static LinkDescriptors serializeLinkDescriptors(final ContainerNode descriptors) {
64         final LinkDescriptorsBuilder linkDescBuilder = new LinkDescriptorsBuilder();
65
66         if (descriptors.getChild(LINK_LOCAL_NID).isPresent() && descriptors.getChild(LINK_REMOTE_NID).isPresent()) {
67             linkDescBuilder.setLinkLocalIdentifier((Uint32) descriptors.getChild(LINK_LOCAL_NID).get().getValue());
68             linkDescBuilder.setLinkRemoteIdentifier((Uint32) descriptors.getChild(LINK_REMOTE_NID).get().getValue());
69         }
70         ifPresentApply(descriptors, IPV4_IFACE_NID,
71             value -> linkDescBuilder.setIpv4InterfaceAddress(new Ipv4InterfaceIdentifier((String) value)));
72         ifPresentApply(descriptors, IPV6_IFACE_NID,
73             value -> linkDescBuilder.setIpv6InterfaceAddress(new Ipv6InterfaceIdentifier((String) value)));
74         ifPresentApply(descriptors, IPV4_NEIGHBOR_NID,
75             value -> linkDescBuilder.setIpv4NeighborAddress(new Ipv4InterfaceIdentifier((String) value)));
76         ifPresentApply(descriptors, IPV6_NEIGHBOR_NID,
77             value -> linkDescBuilder.setIpv6NeighborAddress(new Ipv6InterfaceIdentifier((String) value)));
78         ifPresentApply(descriptors, TlvUtil.MULTI_TOPOLOGY_NID,
79             value -> linkDescBuilder.setMultiTopologyId(new TopologyIdentifier((Integer) value)));
80         return linkDescBuilder.build();
81     }
82
83     private static void ifPresentApply(final ContainerNode descriptors, final NodeIdentifier nid,
84             final SerializerInterface serializer) {
85         if (descriptors.getChild(nid).isPresent()) {
86             serializer.check(descriptors.getChild(nid).get().getValue());
87         }
88     }
89
90     private static LinkDescriptors parseLinkDescriptor(final ByteBuf buffer) {
91         final Map<QName, Object> tlvs = SimpleNlriTypeRegistry.getInstance().parseSubTlvs(buffer);
92         final LinkDescriptorsBuilder builder = new LinkDescriptorsBuilder();
93
94         final LinkLrIdentifiers linkIdentifiers = (LinkLrIdentifiers) tlvs.get(LinkLrIdentifiers.QNAME);
95         if (linkIdentifiers != null) {
96             builder.setLinkLocalIdentifier(linkIdentifiers.getLinkLocalIdentifier());
97             builder.setLinkRemoteIdentifier(linkIdentifiers.getLinkRemoteIdentifier());
98         }
99         builder.setIpv4InterfaceAddress(
100             (Ipv4InterfaceIdentifier) tlvs.get(Ipv4InterfaceTlvParser.IPV4_IFACE_ADDRESS_QNAME));
101         builder.setIpv4NeighborAddress(
102             (Ipv4InterfaceIdentifier) tlvs.get(Ipv4NeighborTlvParser.IPV4_NEIGHBOR_ADDRESS_QNAME));
103         builder.setIpv6InterfaceAddress(
104             (Ipv6InterfaceIdentifier) tlvs.get(Ipv6InterfaceTlvParser.IPV6_IFACE_ADDRESS_QNAME));
105         builder.setIpv6NeighborAddress(
106             (Ipv6InterfaceIdentifier) tlvs.get(Ipv6NeighborTlvParser.IPV6_NEIGHBOR_ADDRESS_QNAME));
107         builder.setMultiTopologyId((TopologyIdentifier) tlvs.get(MultiTopoIdTlvParser.MULTI_TOPOLOGY_ID_QNAME));
108
109         return builder.build();
110     }
111
112     @Override
113     protected ObjectType parseObjectType(final ByteBuf buffer) {
114         final SimpleNlriTypeRegistry reg = SimpleNlriTypeRegistry.getInstance();
115         final LocalNodeDescriptors localDescriptor = reg.parseTlv(buffer);
116         final RemoteNodeDescriptors remoteDescriptor = reg.parseTlv(buffer);
117         final LinkDescriptors linkDescriptor = parseLinkDescriptor(buffer);
118         return new LinkCaseBuilder()
119             .setLinkDescriptors(linkDescriptor)
120             .setLocalNodeDescriptors(localDescriptor)
121             .setRemoteNodeDescriptors(remoteDescriptor)
122             .build();
123     }
124
125     @Override
126     protected void serializeObjectType(final ObjectType objectType, final ByteBuf buffer) {
127         final LinkCase linkNlri = (LinkCase) objectType;
128         final SimpleNlriTypeRegistry reg = SimpleNlriTypeRegistry.getInstance();
129         reg.serializeTlv(LocalNodeDescriptors.QNAME, linkNlri.getLocalNodeDescriptors(), buffer);
130         reg.serializeTlv(RemoteNodeDescriptors.QNAME, linkNlri.getRemoteNodeDescriptors(), buffer);
131         serializeLinkDescriptor(linkNlri.getLinkDescriptors(), buffer);
132     }
133
134     private static void serializeLinkDescriptor(final LinkDescriptors linkDescriptor, final ByteBuf body) {
135         final SimpleNlriTypeRegistry reg = SimpleNlriTypeRegistry.getInstance();
136         if (linkDescriptor.getLinkLocalIdentifier() != null && linkDescriptor.getLinkRemoteIdentifier() != null) {
137             reg.serializeTlv(LinkLrIdentifiers.QNAME, linkDescriptor, body);
138         }
139         reg.serializeTlv(Ipv4InterfaceTlvParser.IPV4_IFACE_ADDRESS_QNAME, linkDescriptor.getIpv4InterfaceAddress(),
140             body);
141         reg.serializeTlv(Ipv4NeighborTlvParser.IPV4_NEIGHBOR_ADDRESS_QNAME, linkDescriptor.getIpv4NeighborAddress(),
142             body);
143         reg.serializeTlv(Ipv6InterfaceTlvParser.IPV6_IFACE_ADDRESS_QNAME, linkDescriptor.getIpv6InterfaceAddress(),
144             body);
145         reg.serializeTlv(Ipv6NeighborTlvParser.IPV6_NEIGHBOR_ADDRESS_QNAME, linkDescriptor.getIpv6NeighborAddress(),
146             body);
147         reg.serializeTlv(MultiTopoIdTlvParser.MULTI_TOPOLOGY_ID_QNAME, linkDescriptor.getMultiTopologyId(), body);
148     }
149
150     @Override
151     public int getNlriType() {
152         return NlriType.Link.getIntValue();
153     }
154 }