17e9b4bc9e025808cb3c9642bd6f9ff5d955dd2c
[bgpcep.git] / bgp / extensions / mvpn / src / main / java / org / opendaylight / protocol / bgp / mvpn / impl / nlri / LeafADHandler.java
1 /*
2  * Copyright (c) 2018 AT&T Intellectual Property. 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.mvpn.impl.nlri;
9
10 import static com.google.common.base.Preconditions.checkArgument;
11
12 import io.netty.buffer.ByteBuf;
13 import io.netty.buffer.Unpooled;
14 import org.opendaylight.bgp.concepts.IpAddressUtil;
15 import org.opendaylight.protocol.bgp.mvpn.spi.pojo.nlri.SimpleMvpnNlriRegistry;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.NlriType;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.leaf.a.d.grouping.LeafAD;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.leaf.a.d.grouping.LeafADBuilder;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.leaf.a.d.grouping.leaf.a.d.LeafADRouteKey;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.leaf.a.d.grouping.leaf.a.d.leaf.a.d.route.key.InterAsIPmsiADCase;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.leaf.a.d.grouping.leaf.a.d.leaf.a.d.route.key.InterAsIPmsiADCaseBuilder;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.leaf.a.d.grouping.leaf.a.d.leaf.a.d.route.key.SPmsiADCase;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.leaf.a.d.grouping.leaf.a.d.leaf.a.d.route.key.SPmsiADCaseBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.mvpn.MvpnChoice;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.mvpn.mvpn.choice.LeafADCase;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.mvpn.mvpn.choice.LeafADCaseBuilder;
27
28 /**
29  * https://tools.ietf.org/html/rfc6514#section-4.4.
30  *
31  * @author Claudio D. Gasparini
32  */
33 public final class LeafADHandler extends AbstractMvpnNlri<LeafADCase> {
34     public LeafADHandler() {
35         super(LeafADCase.class, NlriType.LeafAD);
36     }
37
38     @Override
39     public LeafADCase parseMvpn(final ByteBuf buffer) {
40         final NlriType type = NlriType.forValue(buffer.readUnsignedByte());
41         final short length = buffer.readUnsignedByte();
42         final MvpnChoice key = SimpleMvpnNlriRegistry.getInstance().parseMvpn(type, buffer.readSlice(length));
43         final LeafADRouteKey routeKey;
44         if (type == NlriType.InterAsIPmsiAD) {
45             routeKey = new InterAsIPmsiADCaseBuilder((org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang
46                     .bgp.mvpn.rev200120.mvpn.mvpn.choice.InterAsIPmsiADCase) key).build();
47         } else {
48             routeKey = new SPmsiADCaseBuilder((org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang
49                 .bgp.mvpn.rev200120.mvpn.mvpn.choice.SPmsiADCase) key).build();
50         }
51         return new LeafADCaseBuilder()
52             .setLeafAD(new LeafADBuilder()
53                 .setLeafADRouteKey(routeKey)
54                 .setOrigRouteIp(IpAddressUtil.addressForByteBufWOLength(buffer))
55                 .build())
56             .build();
57     }
58
59     @Override
60     protected ByteBuf serializeBody(final LeafADCase mvpn) {
61         final LeafAD leaf = mvpn.getLeafAD();
62         final ByteBuf nlriByteBuf = Unpooled.buffer();
63         final LeafADRouteKey key = leaf.getLeafADRouteKey();
64         final MvpnChoice keyCase;
65         if (key instanceof InterAsIPmsiADCase) {
66             keyCase = new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang
67                     .bgp.mvpn.rev200120.mvpn.mvpn.choice.InterAsIPmsiADCaseBuilder((InterAsIPmsiADCase) key).build();
68         } else {
69             keyCase = new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev200120.mvpn
70                     .mvpn.choice.SPmsiADCaseBuilder((SPmsiADCase) key).build();
71         }
72         nlriByteBuf.writeBytes(SimpleMvpnNlriRegistry.getInstance().serializeMvpn(keyCase));
73         final ByteBuf orig = IpAddressUtil.bytesWOLengthFor(leaf.getOrigRouteIp());
74         checkArgument(orig.readableBytes() > 0);
75         nlriByteBuf.writeBytes(orig);
76         return nlriByteBuf;
77     }
78 }