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