bgp/rib-impl/src/test/java/org/opendaylight/protocol/bgp/rib/impl/PeerUtil.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.rib.impl;
   9
  10 import java.util.ArrayList;
  11 import java.util.List;
  12 import java.util.Map;
  13 import java.util.stream.Collectors;
  14 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  15 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
  16 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
  17 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv6Prefix;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.prefixes.DestinationIpv4Builder;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.prefixes.destination.ipv4.Ipv4PrefixesBuilder;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv6.prefixes.DestinationIpv6Builder;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv6.prefixes.destination.ipv6.Ipv6PrefixesBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationIpv4CaseBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationIpv6CaseBuilder;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.PathId;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.Update;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.UpdateBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.BgpParameters;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.BgpParametersBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.bgp.parameters.OptionalCapabilities;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.bgp.parameters.OptionalCapabilitiesBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.AttributesBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.AsPath;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.AsPathBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.LocalPref;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.LocalPrefBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.Origin;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.OriginBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.as.path.Segments;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.Attributes1;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.Attributes1Builder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.Attributes2;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.Attributes2Builder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.CParameters1;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.CParameters1Builder;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.SendReceive;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.destination.DestinationType;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.AddPathCapabilityBuilder;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.GracefulRestartCapability;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.GracefulRestartCapabilityBuilder;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.MultiprotocolCapabilityBuilder;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.add.path.capability.AddressFamiliesBuilder;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.graceful.restart.capability.Tables;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.graceful.restart.capability.TablesBuilder;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.update.attributes.MpReachNlri;
  56 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.update.attributes.MpReachNlriBuilder;
  57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.update.attributes.MpUnreachNlri;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.update.attributes.mp.reach.nlri.AdvertizedRoutesBuilder;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
  60 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.AddressFamily;
  61 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.BgpOrigin;
  62 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.Ipv4AddressFamily;
  63 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.Ipv6AddressFamily;
  64 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.UnicastSubsequentAddressFamily;
  65 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.next.hop.CNextHop;
  66 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.next.hop.c.next.hop.Ipv4NextHopCaseBuilder;
  67 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.next.hop.c.next.hop.Ipv6NextHopCaseBuilder;
  68 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.next.hop.c.next.hop.ipv4.next.hop._case.Ipv4NextHopBuilder;
  69 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.next.hop.c.next.hop.ipv6.next.hop._case.Ipv6NextHopBuilder;
  70
  71 final class PeerUtil {
  72
  73     private PeerUtil() {
  74         throw new UnsupportedOperationException();
  75     }
  76
  77     static MpReachNlri createMpReachNlri(final IpAddress nextHop, final long pathId, final List<IpPrefix> prefixes) {
  78         final Class<? extends AddressFamily> afi;
  79         final CNextHop cNextHop;
  80         final DestinationType destinationType;
  81         if (nextHop.getIpv4Address() != null) {
  82             afi = Ipv4AddressFamily.class;
  83             cNextHop = new Ipv4NextHopCaseBuilder().setIpv4NextHop(new Ipv4NextHopBuilder()
  84                     .setGlobal(nextHop.getIpv4Address())
  85                     .build()).build();
  86             destinationType = new DestinationIpv4CaseBuilder().setDestinationIpv4(
  87                     new DestinationIpv4Builder().setIpv4Prefixes(prefixes.stream()
  88                             .map(prefix -> new Ipv4PrefixesBuilder()
  89                                     .setPathId(new PathId(pathId))
  90                                     .setPrefix(new Ipv4Prefix(prefix.getIpv4Prefix())).build())
  91                             .collect(Collectors.toList()))
  92                             .build()).build();
  93         } else {
  94             afi = Ipv6AddressFamily.class;
  95             cNextHop = new Ipv6NextHopCaseBuilder().setIpv6NextHop(new Ipv6NextHopBuilder()
  96                     .setGlobal(nextHop.getIpv6Address())
  97                     .build()).build();
  98             destinationType = new DestinationIpv6CaseBuilder().setDestinationIpv6(
  99                     new DestinationIpv6Builder().setIpv6Prefixes(prefixes.stream()
 100                             .map(prefix -> new Ipv6PrefixesBuilder()
 101                                     .setPathId(new PathId(pathId))
 102                                     .setPrefix(new Ipv6Prefix(prefix.getIpv6Prefix())).build())
 103                             .collect(Collectors.toList()))
 104                             .build()).build();
 105         }
 106
 107         return new MpReachNlriBuilder()
 108                 .setCNextHop(cNextHop)
 109                 .setAfi(afi)
 110                 .setSafi(UnicastSubsequentAddressFamily.class)
 111                 .setAdvertizedRoutes(new AdvertizedRoutesBuilder().setDestinationType(destinationType).build())
 112                 .build();
 113     }
 114
 115     static Update createUpdate(final BgpOrigin bgpOrigin,
 116                                final List<Segments> pathSegments,
 117                                final long preference,
 118                                final MpReachNlri mpReach,
 119                                final MpUnreachNlri mpUnreach) {
 120         final Origin origin = new OriginBuilder().setValue(bgpOrigin).build();
 121         final AsPath asPath = new AsPathBuilder().setSegments(pathSegments).build();
 122         final LocalPref localPref = new LocalPrefBuilder().setPref(preference).build();
 123         final AttributesBuilder attributeBuilder = new AttributesBuilder()
 124                 .setOrigin(origin).setAsPath(asPath).setLocalPref(localPref);
 125
 126         if (mpReach != null) {
 127             attributeBuilder.addAugmentation(Attributes1.class, new Attributes1Builder()
 128                     .setMpReachNlri(mpReach)
 129                     .build());
 130         }
 131
 132         if (mpUnreach != null) {
 133             attributeBuilder.addAugmentation(Attributes2.class, new Attributes2Builder()
 134                     .setMpUnreachNlri(mpUnreach)
 135                     .build());
 136         }
 137
 138         return new UpdateBuilder()
 139                 .setAttributes(new AttributesBuilder()
 140                         .setOrigin(origin)
 141                         .setAsPath(asPath)
 142                         .setLocalPref(localPref)
 143                         .addAugmentation(Attributes1.class, new Attributes1Builder()
 144                                 .setMpReachNlri(mpReach)
 145                                 .build())
 146                         .build()).build();
 147     }
 148
 149     static BgpParameters createBgpParameters(final List<TablesKey> advertisedTables,
 150                                              final List<TablesKey> addPathTables,
 151                                              final Map<TablesKey, Boolean> gracefulTabes,
 152                                              final int gracefulTimer) {
 153         final List<OptionalCapabilities> capabilities = new ArrayList<>();
 154         advertisedTables.forEach(key -> capabilities.add(createMultiprotocolCapability(key)));
 155         if (addPathTables != null && !addPathTables.isEmpty()) {
 156             capabilities.add(createAddPathCapability(addPathTables));
 157         }
 158         if (gracefulTabes != null && !gracefulTabes.isEmpty()) {
 159             capabilities.add(createGracefulRestartCapability(gracefulTabes, gracefulTimer));
 160         }
 161         return new BgpParametersBuilder().setOptionalCapabilities(capabilities).build();
 162     }
 163
 164     private static OptionalCapabilities createMultiprotocolCapability(final TablesKey key) {
 165         return new OptionalCapabilitiesBuilder().setCParameters(
 166                 new CParametersBuilder().addAugmentation(
 167                         CParameters1.class, new CParameters1Builder().setMultiprotocolCapability(
 168                                 new MultiprotocolCapabilityBuilder()
 169                                         .setAfi(key.getAfi())
 170                                         .setSafi(key.getSafi())
 171                                         .build()).build()).build()).build();
 172     }
 173
 174     private static OptionalCapabilities createGracefulRestartCapability(final Map<TablesKey, Boolean> gracefulTables,
 175                                                                         final int restartTime) {
 176         return new OptionalCapabilitiesBuilder().setCParameters(
 177                 new CParametersBuilder().addAugmentation(
 178                         CParameters1.class, new CParameters1Builder()
 179                                 .setGracefulRestartCapability(new GracefulRestartCapabilityBuilder()
 180                                         .setRestartFlags(new GracefulRestartCapability.RestartFlags(false))
 181                                         .setRestartTime(restartTime)
 182                                         .setTables(
 183                                                 gracefulTables.keySet().stream()
 184                                                         .map(key -> new TablesBuilder()
 185                                                                 .setAfi(key.getAfi())
 186                                                                 .setSafi(key.getSafi())
 187                                                                 .setAfiFlags(new Tables.AfiFlags(gracefulTables.get(key)))
 188                                                                 .build())
 189                                                         .collect(Collectors.toList()))
 190                                         .build()).build()).build()).build();
 191     }
 192
 193     private static OptionalCapabilities createAddPathCapability(final List<TablesKey> keys) {
 194         return new OptionalCapabilitiesBuilder().setCParameters(
 195                 new CParametersBuilder().addAugmentation(CParameters1.class,
 196                         new CParameters1Builder().setAddPathCapability(
 197                                 new AddPathCapabilityBuilder().setAddressFamilies(keys.stream()
 198                                         .map(key -> new AddressFamiliesBuilder()
 199                                                 .setAfi(key.getAfi())
 200                                                 .setSafi(key.getSafi())
 201                                                 .setSendReceive(SendReceive.Both)
 202                                                 .build())
 203                                         .collect(Collectors.toList()))
 204                                         .build()).build()).build()).build();
 205     }
 206 }