BUG-4826: BGP Evpn Nlri's handlers

[bgpcep.git] / bgp / evpn / src / main / java / org / opendaylight / protocol / bgp / evpn / impl / nlri / EthSegRParser.java

diff --git a/bgp/evpn/src/main/java/org/opendaylight/protocol/bgp/evpn/impl/nlri/EthSegRParser.java b/bgp/evpn/src/main/java/org/opendaylight/protocol/bgp/evpn/impl/nlri/EthSegRParser.java
new file mode 100644 (file)
index 0000000..0ffe868
--- /dev/null
+++ b/bgp/evpn/src/main/java/org/opendaylight/protocol/bgp/evpn/impl/nlri/EthSegRParser.java
@@ -0,0 +1,104 @@
+/*
+ * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.protocol.bgp.evpn.impl.nlri;
+
+import static org.opendaylight.protocol.bgp.evpn.impl.nlri.NlriModelUtil.extractOrigRouteIp;
+
+import com.google.common.base.Preconditions;
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.Unpooled;
+import org.opendaylight.protocol.bgp.evpn.spi.pojo.SimpleEsiTypeRegistry;
+import org.opendaylight.protocol.util.Ipv4Util;
+import org.opendaylight.protocol.util.Ipv6Util;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.NlriType;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.es.route.EsRoute;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.es.route.EsRouteBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.esi.Esi;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.evpn.EvpnChoice;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.evpn.evpn.choice.EsRouteCase;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.evpn.evpn.choice.EsRouteCaseBuilder;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
+import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
+
+final class EthSegRParser extends AbstractEvpnNlri {
+    protected static final NodeIdentifier ES_ROUTE_NID = new NodeIdentifier(EsRoute.QNAME);
+    private static final int CONTENT_LENGTH = 15;
+    private static final int CONTENT_LENGTH2 = 27;
+
+    @Override
+    public EvpnChoice parseEvpn(final ByteBuf buffer) {
+        Preconditions.checkArgument(buffer.readableBytes() == CONTENT_LENGTH || buffer.readableBytes() == CONTENT_LENGTH2,
+            "Wrong length of array of bytes. Passed: %s ;", buffer);
+
+        final Esi esi = SimpleEsiTypeRegistry.getInstance().parseEsi(buffer.readSlice(ESI_SIZE));
+        final IpAddress ip = Preconditions.checkNotNull(parseOrigRouteIp(buffer));
+
+        final EsRouteBuilder builder = new EsRouteBuilder().setEsi(esi).setOrigRouteIp(ip);
+        return new EsRouteCaseBuilder().setEsRoute(builder.build()).build();
+    }
+
+    @Override
+    protected NlriType getType() {
+        return NlriType.EthSeg;
+    }
+
+    @Override
+    public ByteBuf serializeBody(final EvpnChoice evpnInput) {
+        Preconditions.checkArgument(evpnInput instanceof EsRouteCase, "Unknown evpn instance. Passed %s. Needed EsRouteCase.",
+            evpnInput.getClass());
+        return serializeBody(((EsRouteCase) evpnInput).getEsRoute());
+    }
+
+    @Override
+    public EvpnChoice serializeEvpnModel(final ContainerNode evpnCase) {
+        return createRouteKey(evpnCase);
+    }
+
+    @Override
+    public EvpnChoice createRouteKey(final ContainerNode evpn) {
+        final EsRouteBuilder builder = new EsRouteBuilder();
+        builder.setEsi(serializeEsi(evpn));
+        builder.setOrigRouteIp(extractOrigRouteIp(evpn));
+        return new EsRouteCaseBuilder().setEsRoute(builder.build()).build();
+    }
+
+    private static ByteBuf serializeBody(final EsRoute evpn) {
+        final ByteBuf body = Unpooled.buffer();
+        SimpleEsiTypeRegistry.getInstance().serializeEsi(evpn.getEsi(), body);
+        final ByteBuf orig = serializeOrigRouteIp(evpn.getOrigRouteIp());
+        Preconditions.checkArgument(orig.readableBytes() > 0);
+        body.writeBytes(orig);
+        return body;
+    }
+
+    public static IpAddress parseOrigRouteIp(final ByteBuf buffer) {
+        final int ipLength = buffer.readByte();
+        if (ipLength == Ipv6Util.IPV6_LENGTH) {
+            return new IpAddress(Ipv6Util.addressForByteBuf(buffer));
+        } else if (ipLength == Ipv4Util.IP4_LENGTH) {
+            return new IpAddress(Ipv4Util.addressForByteBuf(buffer));
+        }
+        return null;
+    }
+
+    static ByteBuf serializeOrigRouteIp(final IpAddress origRouteIp) {
+        final ByteBuf body = Unpooled.buffer();
+        if (origRouteIp.getIpv4Address() != null) {
+            body.writeByte(Ipv4Util.IP4_LENGTH);
+            body.writeBytes(Ipv4Util.bytesForAddress(origRouteIp.getIpv4Address()));
+        } else if (origRouteIp.getIpv6Address() != null) {
+            body.writeByte(Ipv6Util.IPV6_LENGTH);
+            body.writeBytes(Ipv6Util.bytesForAddress(origRouteIp.getIpv6Address()));
+        } else {
+            body.writeZero(ZERO_BYTE);
+        }
+        return body;
+    }
+}