bgp/extensions/linkstate/src/main/java/org/opendaylight/protocol/bgp/linkstate/spi/AbstractTeLspNlriCodec.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 package org.opendaylight.protocol.bgp.linkstate.spi;
   9
  10 import com.google.common.annotations.VisibleForTesting;
  11 import com.google.common.base.Preconditions;
  12 import io.netty.buffer.ByteBuf;
  13 import org.opendaylight.protocol.util.ByteBufWriteUtil;
  14 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  15 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv6Address;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.ObjectType;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.destination.CLinkstateDestination;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.TeLspCase;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.TeLspCaseBuilder;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.te.lsp._case.AddressFamily;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.te.lsp._case.address.family.Ipv4Case;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.te.lsp._case.address.family.Ipv4CaseBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.te.lsp._case.address.family.Ipv6Case;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.linkstate.object.type.te.lsp._case.address.family.Ipv6CaseBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.LspId;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.TunnelId;
  27 import org.opendaylight.yangtools.yang.common.QName;
  28 import org.opendaylight.yangtools.yang.common.Uint16;
  29 import org.opendaylight.yangtools.yang.common.Uint32;
  30 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  31 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
  32
  33 public abstract class AbstractTeLspNlriCodec extends AbstractNlriTypeCodec {
  34
  35     @VisibleForTesting
  36     public static final NodeIdentifier LSP_ID = NodeIdentifier.create(
  37         QName.create(CLinkstateDestination.QNAME, "lsp-id").intern());
  38     @VisibleForTesting
  39     public static final NodeIdentifier TUNNEL_ID = NodeIdentifier.create(
  40         QName.create(CLinkstateDestination.QNAME, "tunnel-id").intern());
  41     @VisibleForTesting
  42     public static final NodeIdentifier IPV4_TUNNEL_SENDER_ADDRESS = NodeIdentifier.create(
  43         QName.create(CLinkstateDestination.QNAME, "ipv4-tunnel-sender-address").intern());
  44     @VisibleForTesting
  45     public static final NodeIdentifier IPV4_TUNNEL_ENDPOINT_ADDRESS = NodeIdentifier.create(
  46         QName.create(CLinkstateDestination.QNAME, "ipv4-tunnel-endpoint-address").intern());
  47     @VisibleForTesting
  48     private static final NodeIdentifier IPV6_TUNNEL_SENDER_ADDRESS = NodeIdentifier.create(
  49         QName.create(CLinkstateDestination.QNAME, "ipv6-tunnel-sender-address").intern());
  50     @VisibleForTesting
  51     private static final NodeIdentifier IPV6_TUNNEL_ENDPOINT_ADDRESS = NodeIdentifier.create(
  52         QName.create(CLinkstateDestination.QNAME, "ipv6-tunnel-endpoint-address").intern());
  53
  54     @VisibleForTesting
  55     public static final NodeIdentifier ADDRESS_FAMILY = NodeIdentifier.create(AddressFamily.QNAME);
  56
  57     public static boolean isTeLsp(final ChoiceNode objectType) {
  58         return objectType.getChild(ADDRESS_FAMILY).isPresent();
  59     }
  60
  61     public static TeLspCase serializeTeLsp(final ChoiceNode objectType) {
  62         final TeLspCaseBuilder teLsp = new TeLspCaseBuilder();
  63         teLsp.setLspId(new LspId((Uint32) objectType.getChild(LSP_ID).get().getValue()));
  64         teLsp.setTunnelId(new TunnelId((Uint16) objectType.getChild(TUNNEL_ID).get().getValue()));
  65         final ChoiceNode addressFamily = (ChoiceNode) objectType.getChild(ADDRESS_FAMILY).get();
  66         teLsp.setAddressFamily(serializeAddressFamily(addressFamily,
  67             addressFamily.getChild(IPV4_TUNNEL_SENDER_ADDRESS).isPresent()));
  68
  69         return teLsp.build();
  70     }
  71
  72     private static AddressFamily serializeAddressFamily(final ChoiceNode addressFamily, final boolean ipv4Case) {
  73         if (ipv4Case) {
  74             return new Ipv4CaseBuilder()
  75                 .setIpv4TunnelSenderAddress(
  76                     new Ipv4Address((String) addressFamily.getChild(IPV4_TUNNEL_SENDER_ADDRESS).get().getValue()))
  77                 .setIpv4TunnelEndpointAddress(
  78                     new Ipv4Address((String) addressFamily.getChild(IPV4_TUNNEL_ENDPOINT_ADDRESS).get().getValue()))
  79                 .build();
  80         }
  81
  82         return new Ipv6CaseBuilder()
  83             .setIpv6TunnelSenderAddress(
  84                 new Ipv6Address((String) addressFamily.getChild(IPV6_TUNNEL_SENDER_ADDRESS).get().getValue()))
  85             .setIpv6TunnelEndpointAddress(
  86                 new Ipv6Address((String) addressFamily.getChild(IPV6_TUNNEL_ENDPOINT_ADDRESS).get().getValue()))
  87             .build();
  88     }
  89
  90     @Override
  91     protected final void serializeObjectType(final ObjectType objectType, final ByteBuf buffer) {
  92         Preconditions.checkArgument(objectType instanceof TeLspCase);
  93         final TeLspCase teLSP = (TeLspCase) objectType;
  94         final AddressFamily addressFamily = teLSP.getAddressFamily();
  95         if (addressFamily instanceof Ipv4Case) {
  96             serializeIpv4Case(teLSP, (Ipv4Case) addressFamily, buffer);
  97         } else {
  98             serializeIpv6Case(teLSP, (Ipv6Case) addressFamily, buffer);
  99         }
 100     }
 101
 102     private static void serializeIpv4Case(final TeLspCase teLSP, final Ipv4Case ipv4Case, final ByteBuf buffer) {
 103         ByteBufWriteUtil.writeIpv4Address(ipv4Case.getIpv4TunnelSenderAddress(), buffer);
 104         serializeTunnelIdAndLspId(buffer, teLSP);
 105         ByteBufWriteUtil.writeIpv4Address(ipv4Case.getIpv4TunnelEndpointAddress(), buffer);
 106     }
 107
 108     private static void serializeIpv6Case(final TeLspCase teLSP, final Ipv6Case ipv6Case, final ByteBuf buffer) {
 109         ByteBufWriteUtil.writeIpv6Address(ipv6Case.getIpv6TunnelSenderAddress(), buffer);
 110         serializeTunnelIdAndLspId(buffer, teLSP);
 111         ByteBufWriteUtil.writeIpv6Address(ipv6Case.getIpv6TunnelEndpointAddress(), buffer);
 112     }
 113
 114     private static void serializeTunnelIdAndLspId(final ByteBuf buffer, final TeLspCase teLSP) {
 115         ByteBufWriteUtil.writeUnsignedShort(teLSP.getTunnelId().getValue(), buffer);
 116         ByteBufWriteUtil.writeUnsignedShort(teLSP.getLspId().getValue().intValue(), buffer);
 117     }
 118 }