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.data.api.YangInstanceIdentifier.NodeIdentifier;
  29 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
  30
  31 public abstract class AbstractTeLspNlriCodec extends AbstractNlriTypeCodec {
  32
  33     @VisibleForTesting
  34     public static final NodeIdentifier LSP_ID = NodeIdentifier.create(
  35         QName.create(CLinkstateDestination.QNAME, "lsp-id").intern());
  36     @VisibleForTesting
  37     public static final NodeIdentifier TUNNEL_ID = NodeIdentifier.create(
  38         QName.create(CLinkstateDestination.QNAME, "tunnel-id").intern());
  39     @VisibleForTesting
  40     public static final NodeIdentifier IPV4_TUNNEL_SENDER_ADDRESS = NodeIdentifier.create(
  41         QName.create(CLinkstateDestination.QNAME, "ipv4-tunnel-sender-address").intern());
  42     @VisibleForTesting
  43     public static final NodeIdentifier IPV4_TUNNEL_ENDPOINT_ADDRESS = NodeIdentifier.create(
  44         QName.create(CLinkstateDestination.QNAME, "ipv4-tunnel-endpoint-address").intern());
  45     @VisibleForTesting
  46     private static final NodeIdentifier IPV6_TUNNEL_SENDER_ADDRESS = NodeIdentifier.create(
  47         QName.create(CLinkstateDestination.QNAME, "ipv6-tunnel-sender-address").intern());
  48     @VisibleForTesting
  49     private static final NodeIdentifier IPV6_TUNNEL_ENDPOINT_ADDRESS = NodeIdentifier.create(
  50         QName.create(CLinkstateDestination.QNAME, "ipv6-tunnel-endpoint-address").intern());
  51
  52     @VisibleForTesting
  53     public static final NodeIdentifier ADDRESS_FAMILY = NodeIdentifier.create(AddressFamily.QNAME);
  54
  55     public static boolean isTeLsp(final ChoiceNode objectType) {
  56         return objectType.getChild(ADDRESS_FAMILY).isPresent();
  57     }
  58
  59     public static TeLspCase serializeTeLsp(final ChoiceNode objectType) {
  60         final TeLspCaseBuilder teLsp = new TeLspCaseBuilder();
  61         teLsp.setLspId(new LspId((Long) objectType.getChild(LSP_ID).get().getValue()));
  62         teLsp.setTunnelId(new TunnelId((Integer) objectType.getChild(TUNNEL_ID).get().getValue()));
  63         final ChoiceNode addressFamily = (ChoiceNode) objectType.getChild(ADDRESS_FAMILY).get();
  64         teLsp.setAddressFamily(serializeAddressFamily(addressFamily,
  65             addressFamily.getChild(IPV4_TUNNEL_SENDER_ADDRESS).isPresent()));
  66
  67         return teLsp.build();
  68     }
  69
  70     private static AddressFamily serializeAddressFamily(final ChoiceNode addressFamily, final boolean ipv4Case) {
  71         if (ipv4Case) {
  72             return new Ipv4CaseBuilder()
  73                 .setIpv4TunnelSenderAddress(
  74                     new Ipv4Address((String) addressFamily.getChild(IPV4_TUNNEL_SENDER_ADDRESS).get().getValue()))
  75                 .setIpv4TunnelEndpointAddress(
  76                     new Ipv4Address((String) addressFamily.getChild(IPV4_TUNNEL_ENDPOINT_ADDRESS).get().getValue()))
  77                 .build();
  78         }
  79
  80         return new Ipv6CaseBuilder()
  81             .setIpv6TunnelSenderAddress(
  82                 new Ipv6Address((String) addressFamily.getChild(IPV6_TUNNEL_SENDER_ADDRESS).get().getValue()))
  83             .setIpv6TunnelEndpointAddress(
  84                 new Ipv6Address((String) addressFamily.getChild(IPV6_TUNNEL_ENDPOINT_ADDRESS).get().getValue()))
  85             .build();
  86     }
  87
  88     @Override
  89     protected final void serializeObjectType(final ObjectType objectType, final ByteBuf buffer) {
  90         Preconditions.checkArgument(objectType instanceof TeLspCase);
  91         final TeLspCase teLSP = (TeLspCase) objectType;
  92         final AddressFamily addressFamily = teLSP.getAddressFamily();
  93         if (addressFamily instanceof Ipv4Case) {
  94             serializeIpv4Case(teLSP, (Ipv4Case) addressFamily, buffer);
  95         } else {
  96             serializeIpv6Case(teLSP, (Ipv6Case) addressFamily, buffer);
  97         }
  98     }
  99
 100     private static void serializeIpv4Case(final TeLspCase teLSP, final Ipv4Case ipv4Case, final ByteBuf buffer) {
 101         ByteBufWriteUtil.writeIpv4Address(ipv4Case.getIpv4TunnelSenderAddress(), buffer);
 102         serializeTunnelIdAndLspId(buffer, teLSP);
 103         ByteBufWriteUtil.writeIpv4Address(ipv4Case.getIpv4TunnelEndpointAddress(), buffer);
 104     }
 105
 106     private static void serializeIpv6Case(final TeLspCase teLSP, final Ipv6Case ipv6Case, final ByteBuf buffer) {
 107         ByteBufWriteUtil.writeIpv6Address(ipv6Case.getIpv6TunnelSenderAddress(), buffer);
 108         serializeTunnelIdAndLspId(buffer, teLSP);
 109         ByteBufWriteUtil.writeIpv6Address(ipv6Case.getIpv6TunnelEndpointAddress(), buffer);
 110     }
 111
 112     private static void serializeTunnelIdAndLspId(final ByteBuf buffer, final TeLspCase teLSP) {
 113         ByteBufWriteUtil.writeUnsignedShort(teLSP.getTunnelId().getValue(), buffer);
 114         ByteBufWriteUtil.writeUnsignedShort(teLSP.getLspId().getValue().intValue(), buffer);
 115     }
 116 }