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
   9 package org.opendaylight.protocol.bgp.mvpn.impl.nlri;
  10
  11 import com.google.common.base.Preconditions;
  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.rev180417.NlriType;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.leaf.a.d.grouping.LeafAD;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.leaf.a.d.grouping.LeafADBuilder;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.leaf.a.d.grouping.leaf.a.d.LeafADRouteKey;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.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.rev180417.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.rev180417.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.rev180417.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.rev180417.mvpn.MvpnChoice;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.mvpn.mvpn.choice.LeafADCase;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.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     @Override
  35     public int getType() {
  36         return NlriType.LeafAD.getIntValue();
  37     }
  38
  39     @Override
  40     public LeafADCase parseMvpn(final ByteBuf buffer) {
  41         final NlriType type = NlriType.forValue(buffer.readUnsignedByte());
  42         final short length = buffer.readUnsignedByte();
  43         final MvpnChoice key = SimpleMvpnNlriRegistry.getInstance().parseMvpn(type, buffer.readBytes(length));
  44         final LeafADRouteKey routeKey;
  45         if (type == NlriType.InterAsIPmsiAD) {
  46             routeKey = new InterAsIPmsiADCaseBuilder((org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang
  47                     .bgp.mvpn.rev180417.mvpn.mvpn.choice.InterAsIPmsiADCase) key).build();
  48         } else {
  49             routeKey = new SPmsiADCaseBuilder((org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp
  50                     .mvpn.rev180417.mvpn.mvpn.choice.SPmsiADCase) key).build();
  51         }
  52         return new LeafADCaseBuilder().setLeafAD(new LeafADBuilder()
  53                 .setLeafADRouteKey(routeKey)
  54                 .setOrigRouteIp(IpAddressUtil.addressForByteBufWOLength(buffer)).build())
  55                 .build();
  56     }
  57
  58     @Override
  59     protected ByteBuf serializeBody(final LeafADCase mvpn) {
  60         final LeafAD leaf = mvpn.getLeafAD();
  61         final ByteBuf nlriByteBuf = Unpooled.buffer();
  62         final LeafADRouteKey key = leaf.getLeafADRouteKey();
  63         final MvpnChoice keyCase;
  64         if (key instanceof InterAsIPmsiADCase) {
  65             keyCase = new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang
  66                     .bgp.mvpn.rev180417.mvpn.mvpn.choice.InterAsIPmsiADCaseBuilder((InterAsIPmsiADCase) key).build();
  67         } else {
  68             keyCase = new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.mvpn
  69                     .mvpn.choice.SPmsiADCaseBuilder((SPmsiADCase) key).build();
  70         }
  71         nlriByteBuf.writeBytes(SimpleMvpnNlriRegistry.getInstance().serializeMvpn(keyCase));
  72         final ByteBuf orig = IpAddressUtil.bytesWOLengthFor(leaf.getOrigRouteIp());
  73         Preconditions.checkArgument(orig.readableBytes() > 0);
  74         nlriByteBuf.writeBytes(orig);
  75         return nlriByteBuf;
  76     }
  77
  78     @Override
  79     public Class<? extends MvpnChoice> getClazz() {
  80         return LeafADCase.class;
  81     }
  82 }