BUG-5983: Fix wrong representation of Ip length
[bgpcep.git] / bgp / evpn / src / main / java / org / opendaylight / protocol / bgp / evpn / impl / nlri / EthSegRParser.java
1 /*
2  * Copyright (c) 2016 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
9 package org.opendaylight.protocol.bgp.evpn.impl.nlri;
10
11 import static org.opendaylight.protocol.bgp.evpn.impl.nlri.NlriModelUtil.extractOrigRouteIp;
12
13 import com.google.common.base.Preconditions;
14 import io.netty.buffer.ByteBuf;
15 import io.netty.buffer.Unpooled;
16 import org.opendaylight.protocol.bgp.evpn.spi.pojo.SimpleEsiTypeRegistry;
17 import org.opendaylight.protocol.util.Ipv4Util;
18 import org.opendaylight.protocol.util.Ipv6Util;
19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.NlriType;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.es.route.EsRoute;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.es.route.EsRouteBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.esi.Esi;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.evpn.EvpnChoice;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.evpn.evpn.choice.EsRouteCase;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev160321.evpn.evpn.choice.EsRouteCaseBuilder;
27 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
28 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
29
30 final class EthSegRParser extends AbstractEvpnNlri {
31     protected static final NodeIdentifier ES_ROUTE_NID = new NodeIdentifier(EsRoute.QNAME);
32     private static final int CONTENT_LENGTH = 15;
33     private static final int CONTENT_LENGTH2 = 27;
34
35     @Override
36     public EvpnChoice parseEvpn(final ByteBuf buffer) {
37         Preconditions.checkArgument(buffer.readableBytes() == CONTENT_LENGTH || buffer.readableBytes() == CONTENT_LENGTH2,
38             "Wrong length of array of bytes. Passed: %s ;", buffer);
39
40         final Esi esi = SimpleEsiTypeRegistry.getInstance().parseEsi(buffer.readSlice(ESI_SIZE));
41         final IpAddress ip = Preconditions.checkNotNull(parseOrigRouteIp(buffer));
42
43         final EsRouteBuilder builder = new EsRouteBuilder().setEsi(esi).setOrigRouteIp(ip);
44         return new EsRouteCaseBuilder().setEsRoute(builder.build()).build();
45     }
46
47     @Override
48     protected NlriType getType() {
49         return NlriType.EthSeg;
50     }
51
52     @Override
53     public ByteBuf serializeBody(final EvpnChoice evpnInput) {
54         Preconditions.checkArgument(evpnInput instanceof EsRouteCase, "Unknown evpn instance. Passed %s. Needed EsRouteCase.",
55             evpnInput.getClass());
56         return serializeBody(((EsRouteCase) evpnInput).getEsRoute());
57     }
58
59     @Override
60     public EvpnChoice serializeEvpnModel(final ContainerNode evpnCase) {
61         return createRouteKey(evpnCase);
62     }
63
64     @Override
65     public EvpnChoice createRouteKey(final ContainerNode evpn) {
66         final EsRouteBuilder builder = new EsRouteBuilder();
67         builder.setEsi(serializeEsi(evpn));
68         builder.setOrigRouteIp(extractOrigRouteIp(evpn));
69         return new EsRouteCaseBuilder().setEsRoute(builder.build()).build();
70     }
71
72     private static ByteBuf serializeBody(final EsRoute evpn) {
73         final ByteBuf body = Unpooled.buffer();
74         SimpleEsiTypeRegistry.getInstance().serializeEsi(evpn.getEsi(), body);
75         final ByteBuf orig = serializeOrigRouteIp(evpn.getOrigRouteIp());
76         Preconditions.checkArgument(orig.readableBytes() > 0);
77         body.writeBytes(orig);
78         return body;
79     }
80
81     public static IpAddress parseOrigRouteIp(final ByteBuf buffer) {
82         final int ipLength = buffer.readUnsignedByte();
83         if (ipLength == Ipv6Util.IPV6_BITS_LENGTH) {
84             return new IpAddress(Ipv6Util.addressForByteBuf(buffer));
85         } else if (ipLength == Ipv4Util.IP4_BITS_LENGTH) {
86             return new IpAddress(Ipv4Util.addressForByteBuf(buffer));
87         }
88         return null;
89     }
90
91     static ByteBuf serializeOrigRouteIp(final IpAddress origRouteIp) {
92         final ByteBuf body = Unpooled.buffer();
93         if (origRouteIp.getIpv4Address() != null) {
94             body.writeByte(Ipv4Util.IP4_BITS_LENGTH);
95             body.writeBytes(Ipv4Util.bytesForAddress(origRouteIp.getIpv4Address()));
96         } else if (origRouteIp.getIpv6Address() != null) {
97             body.writeByte(Ipv6Util.IPV6_BITS_LENGTH);
98             body.writeBytes(Ipv6Util.bytesForAddress(origRouteIp.getIpv6Address()));
99         } else {
100             body.writeZero(ZERO_BYTE);
101         }
102         return body;
103     }
104 }