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.common.netty.ByteBufUtils;
  31 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  32 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
  33
  34 public abstract class AbstractTeLspNlriCodec extends AbstractNlriTypeCodec {
  35
  36     @VisibleForTesting
  37     public static final NodeIdentifier LSP_ID = NodeIdentifier.create(
  38         QName.create(CLinkstateDestination.QNAME, "lsp-id").intern());
  39     @VisibleForTesting
  40     public static final NodeIdentifier TUNNEL_ID = NodeIdentifier.create(
  41         QName.create(CLinkstateDestination.QNAME, "tunnel-id").intern());
  42     @VisibleForTesting
  43     public static final NodeIdentifier IPV4_TUNNEL_SENDER_ADDRESS = NodeIdentifier.create(
  44         QName.create(CLinkstateDestination.QNAME, "ipv4-tunnel-sender-address").intern());
  45     @VisibleForTesting
  46     public static final NodeIdentifier IPV4_TUNNEL_ENDPOINT_ADDRESS = NodeIdentifier.create(
  47         QName.create(CLinkstateDestination.QNAME, "ipv4-tunnel-endpoint-address").intern());
  48     @VisibleForTesting
  49     private static final NodeIdentifier IPV6_TUNNEL_SENDER_ADDRESS = NodeIdentifier.create(
  50         QName.create(CLinkstateDestination.QNAME, "ipv6-tunnel-sender-address").intern());
  51     @VisibleForTesting
  52     private static final NodeIdentifier IPV6_TUNNEL_ENDPOINT_ADDRESS = NodeIdentifier.create(
  53         QName.create(CLinkstateDestination.QNAME, "ipv6-tunnel-endpoint-address").intern());
  54
  55     @VisibleForTesting
  56     public static final NodeIdentifier ADDRESS_FAMILY = NodeIdentifier.create(AddressFamily.QNAME);
  57
  58     public static boolean isTeLsp(final ChoiceNode objectType) {
  59         return objectType.getChild(ADDRESS_FAMILY).isPresent();
  60     }
  61
  62     public static TeLspCase serializeTeLsp(final ChoiceNode objectType) {
  63         final ChoiceNode addressFamily = (ChoiceNode) objectType.getChild(ADDRESS_FAMILY).get();
  64         return new TeLspCaseBuilder()
  65                 .setLspId(new LspId((Uint32) objectType.getChild(LSP_ID).get().getValue()))
  66                 .setTunnelId(new TunnelId((Uint16) objectType.getChild(TUNNEL_ID).get().getValue()))
  67                 .setAddressFamily(serializeAddressFamily(addressFamily,
  68                     addressFamily.getChild(IPV4_TUNNEL_SENDER_ADDRESS).isPresent()))
  69                 .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         ByteBufUtils.write(buffer, teLSP.getTunnelId().getValue());
 116         buffer.writeShort(teLSP.getLspId().getValue().intValue());
 117     }
 118 }