--- /dev/null
+/*
+ * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.protocol.bgp.rib.impl;
+
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.doReturn;
+import static org.opendaylight.protocol.bgp.rib.spi.RouterIds.createPeerId;
+
+import com.google.common.base.Optional;
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.Lists;
+import com.google.common.net.InetAddresses;
+import io.netty.bootstrap.Bootstrap;
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.ByteBufUtil;
+import io.netty.channel.Channel;
+import io.netty.channel.ChannelFuture;
+import io.netty.channel.ChannelHandlerContext;
+import io.netty.channel.ChannelInitializer;
+import io.netty.channel.ChannelOption;
+import io.netty.channel.SimpleChannelInboundHandler;
+import io.netty.channel.epoll.Epoll;
+import io.netty.channel.epoll.EpollEventLoopGroup;
+import io.netty.channel.nio.NioEventLoopGroup;
+import io.netty.channel.socket.SocketChannel;
+import io.netty.util.concurrent.DefaultPromise;
+import java.net.InetAddress;
+import java.net.InetSocketAddress;
+import java.util.Collections;
+import java.util.List;
+import java.util.concurrent.ExecutionException;
+import javassist.ClassPool;
+import org.junit.After;
+import org.junit.Assert;
+import org.junit.Before;
+import org.mockito.Mockito;
+import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
+import org.opendaylight.controller.md.sal.binding.impl.BindingToNormalizedNodeCodec;
+import org.opendaylight.controller.md.sal.binding.test.AbstractDataBrokerTest;
+import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
+import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonService;
+import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider;
+import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceRegistration;
+import org.opendaylight.protocol.bgp.inet.RIBActivator;
+import org.opendaylight.protocol.bgp.parser.impl.BGPActivator;
+import org.opendaylight.protocol.bgp.parser.spi.BGPExtensionProviderContext;
+import org.opendaylight.protocol.bgp.parser.spi.pojo.SimpleBGPExtensionProviderContext;
+import org.opendaylight.protocol.bgp.rib.impl.spi.BGPPeerRegistry;
+import org.opendaylight.protocol.bgp.rib.impl.spi.BGPSessionPreferences;
+import org.opendaylight.protocol.bgp.rib.spi.BGPSessionListener;
+import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionProviderContext;
+import org.opendaylight.protocol.bgp.rib.spi.SimpleRIBExtensionProviderContext;
+import org.opendaylight.protocol.concepts.KeyMapping;
+import org.opendaylight.protocol.util.InetSocketAddressUtil;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.bgp.rib.rib.loc.rib.tables.routes.Ipv4RoutesCase;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.ipv4.prefixes.DestinationIpv4Builder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.ipv4.prefixes.destination.ipv4.Ipv4PrefixesBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.ipv4.routes.ipv4.routes.Ipv4Route;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationIpv4Case;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationIpv4CaseBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.PathId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Update;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.UpdateBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParameters;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParametersBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.OptionalCapabilities;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.OptionalCapabilitiesBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.AttributesBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.AsPathBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.ClusterIdBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.LocalPrefBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.OriginBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.attributes.OriginatorIdBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.Attributes1;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.Attributes1Builder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1Builder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.SendReceive;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.AddPathCapabilityBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.MultiprotocolCapability;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.MultiprotocolCapabilityBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.add.path.capability.AddressFamiliesBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.MpReachNlri;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.MpReachNlriBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.mp.reach.nlri.AdvertizedRoutes;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.mp.reach.nlri.AdvertizedRoutesBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.BgpRib;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerRole;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.Peer;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpOrigin;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.ClusterIdentifier;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.Ipv4NextHopCaseBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.next.hop.c.next.hop.ipv4.next.hop._case.Ipv4NextHopBuilder;
+import org.opendaylight.yangtools.binding.data.codec.gen.impl.StreamWriterGenerator;
+import org.opendaylight.yangtools.binding.data.codec.impl.BindingNormalizedNodeCodecRegistry;
+import org.opendaylight.yangtools.sal.binding.generator.impl.GeneratedClassLoadingStrategy;
+import org.opendaylight.yangtools.sal.binding.generator.impl.ModuleInfoBackedContext;
+import org.opendaylight.yangtools.sal.binding.generator.util.JavassistUtils;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.opendaylight.yangtools.yang.binding.util.BindingReflections;
+import org.opendaylight.yangtools.yang.model.api.SchemaContext;
+
+public class AbstractAddPathTest extends AbstractDataBrokerTest {
+
+ protected static final String RIB_ID = "127.0.0.1";
+ private static final ClusterIdentifier CLUSTER_ID = new ClusterIdentifier(RIB_ID);
+ protected static final Ipv4Address PEER1 = new Ipv4Address("127.0.0.2");
+ protected static final Ipv4Address PEER2 = new Ipv4Address("127.0.0.3");
+ protected static final Ipv4Address PEER3 = new Ipv4Address("127.0.0.4");
+ protected static final Ipv4Address PEER4 = new Ipv4Address("127.0.0.5");
+ protected static final Ipv4Address PEER5 = new Ipv4Address("127.0.0.6");
+
+ protected static final PeerId PEER1_ID = createPeerId(PEER1);
+ protected static final PeerId PEER2_ID = createPeerId(PEER2);
+ protected static final PeerId PEER3_ID = createPeerId(PEER3);
+ protected static final PeerId PEER4_ID = createPeerId(PEER4);
+ protected static final PeerId PEER5_ID = createPeerId(PEER5);
+
+ protected static final AsNumber AS_NUMBER = new AsNumber(72L);
+ protected static final int HOLDTIMER = 180;
+
+ protected static final int PORT = InetSocketAddressUtil.getRandomPort();
+
+ protected static final Ipv4Prefix PREFIX1 = new Ipv4Prefix("1.1.1.1/32");
+ protected static final Ipv4Address NH1 = new Ipv4Address("2.2.2.2");
+ protected static final Update UPD_100 = createSimpleUpdate(PREFIX1, new PathId(1l), CLUSTER_ID, 100);
+ protected static final Update UPD_50 = createSimpleUpdate(PREFIX1, new PathId(2l), CLUSTER_ID, 50);
+ protected static final Update UPD_200 = createSimpleUpdate(PREFIX1, new PathId(3l), CLUSTER_ID, 200);
+ protected static final Update UPD_20 = createSimpleUpdate(PREFIX1, new PathId(1l), CLUSTER_ID, 20);
+
+ protected BindingToNormalizedNodeCodec mappingService;
+ protected BGPDispatcherImpl dispatcher;
+ protected RIBExtensionProviderContext ribExtension;
+ protected BGPExtensionProviderContext context;
+ protected SchemaContext schemaContext;
+ private RIBActivator ribActivator;
+ private BGPActivator bgpActivator;
+ private NioEventLoopGroup worker;
+ private NioEventLoopGroup boss;
+ protected ClusterSingletonServiceProvider clusterSingletonServiceProvider;
+ private org.opendaylight.protocol.bgp.inet.BGPActivator inetActivator;
+
+ @Before
+ public void setUp() throws Exception {
+ this.ribActivator = new RIBActivator();
+ this.ribExtension = new SimpleRIBExtensionProviderContext();
+
+ this.ribActivator.startRIBExtensionProvider(this.ribExtension);
+
+ this.bgpActivator = new BGPActivator();
+ this.inetActivator = new org.opendaylight.protocol.bgp.inet.BGPActivator();
+ this.context = new SimpleBGPExtensionProviderContext();
+ this.bgpActivator.start(this.context);
+ this.inetActivator.start(this.context);
+
+ this.mappingService = new BindingToNormalizedNodeCodec(GeneratedClassLoadingStrategy.getTCCLClassLoadingStrategy(),
+ new BindingNormalizedNodeCodecRegistry(StreamWriterGenerator.create(JavassistUtils.forClassPool(ClassPool.getDefault()))));
+ final ModuleInfoBackedContext moduleInfoBackedContext = ModuleInfoBackedContext.create();
+ moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(BgpParameters.class));
+ moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(MultiprotocolCapability.class));
+ moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(DestinationIpv4Case.class));
+ moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(AdvertizedRoutes.class));
+ moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(BgpRib.class));
+ moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(Attributes1.class));
+ moduleInfoBackedContext.registerModuleInfo(BindingReflections.getModuleInfo(MpReachNlri.class));
+ this.mappingService.onGlobalContextUpdated(moduleInfoBackedContext.tryToCreateSchemaContext().get());
+ this.schemaContext = moduleInfoBackedContext.getSchemaContext();
+
+ this.worker = new NioEventLoopGroup();
+ this.boss = new NioEventLoopGroup();
+ this.dispatcher = new BGPDispatcherImpl(this.context.getMessageRegistry(), this.boss, this.worker);
+
+ this.clusterSingletonServiceProvider = Mockito.mock(ClusterSingletonServiceProvider.class);
+ doReturn(Mockito.mock(ClusterSingletonServiceRegistration.class)).when(this.clusterSingletonServiceProvider).registerClusterSingletonService(any(ClusterSingletonService.class));
+ }
+
+ @After
+ public void tearDown() {
+ this.dispatcher.close();
+ this.worker.shutdownGracefully().awaitUninterruptibly();
+ this.boss.shutdownGracefully().awaitUninterruptibly();
+ this.mappingService.close();
+ this.ribActivator.close();
+ this.inetActivator.close();
+ this.bgpActivator.close();
+ }
+
+ void checkRibOut(final int nAddPathRoutesExpected) throws ExecutionException, InterruptedException {
+ final ReadOnlyTransaction rTx = getDataBroker().newReadOnlyTransaction();
+ final BgpRib bgpRib = rTx.read(LogicalDatastoreType.OPERATIONAL, InstanceIdentifier.create(BgpRib.class)).get().get();
+ rTx.close();
+
+ //check peer's rib-out
+ for (final Peer peer : bgpRib.getRib().get(0).getPeer()) {
+ final int ribOut = getPeerRibOutSize(peer);
+ if (peer.getPeerId().equals(PEER1_ID)) {
+ Assert.assertEquals(0, ribOut);
+ } else if (peer.getPeerId().equals(PEER2_ID)) {
+ Assert.assertEquals(0, ribOut);
+ } else if (peer.getPeerId().equals(PEER3_ID)) {
+ Assert.assertEquals(0, ribOut);
+ } else if (peer.getPeerId().equals(PEER4_ID)) {
+ Assert.assertEquals(1, ribOut);
+ } else if (peer.getPeerId().equals(PEER5_ID)) {
+ Assert.assertEquals(nAddPathRoutesExpected, ribOut);
+ } else {
+ Assert.fail("Failed to verify " + peer);
+ }
+ }
+ }
+
+ void sendRouteAndCheckIsOnDS(final Channel session, final Ipv4Prefix prefix, final long localPreference, final int expectedRoutesOnDS)
+ throws InterruptedException, ExecutionException {
+ session.writeAndFlush(createSimpleUpdate(prefix, null, null, localPreference));
+ Thread.sleep(2000);
+
+ final ReadOnlyTransaction rTx = getDataBroker().newReadOnlyTransaction();
+ final BgpRib bgpRib = rTx.read(LogicalDatastoreType.OPERATIONAL, InstanceIdentifier.create(BgpRib.class)).get().get();
+ rTx.close();
+ final Ipv4RoutesCase routes = ((Ipv4RoutesCase) bgpRib.getRib().get(0).getLocRib().getTables().get(0).getRoutes());
+ final List<Ipv4Route> routeList = routes.getIpv4Routes().getIpv4Route();
+ Assert.assertEquals(expectedRoutesOnDS, routeList.size());
+ }
+
+ void checkPeersPresentOnDataStore(final int numberOfPeers) throws ExecutionException, InterruptedException {
+ Thread.sleep(1000);
+ final ReadOnlyTransaction rTx = getDataBroker().newReadOnlyTransaction();
+ final BgpRib bgpRib = rTx.read(LogicalDatastoreType.OPERATIONAL, InstanceIdentifier.create(BgpRib.class)).get().get();
+ rTx.close();
+
+ //check 5 peers present in the DS
+ Assert.assertEquals(numberOfPeers, bgpRib.getRib().get(0).getPeer().size());
+ }
+
+ Channel createPeerSession(final Ipv4Address peer, final PeerRole peerRole, final BgpParameters nonAddPathParams, final RIBImpl ribImpl,
+ final BGPHandlerFactory hf, final SimpleSessionListener sessionListsner) throws InterruptedException, ExecutionException {
+ configurePeer(peer, ribImpl, nonAddPathParams, peerRole);
+ return connectPeer(peer, ribImpl, nonAddPathParams, this.dispatcher, hf, sessionListsner);
+ }
+
+ private static int getPeerRibOutSize(final Peer peer) {
+ return ((org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev150305.bgp.rib.rib.peer.adj.rib.out.tables.routes.Ipv4RoutesCase) peer.getAdjRibOut().getTables().get(0).getRoutes()).getIpv4Routes().getIpv4Route().size();
+ }
+
+ private static ChannelFuture createClient(final BGPDispatcherImpl dispatcher, final InetSocketAddress remoteAddress,
+ final BGPPeerRegistry registry, final InetSocketAddress localAddress, final BGPHandlerFactory hf) throws InterruptedException {
+ final BGPClientSessionNegotiatorFactory snf = new BGPClientSessionNegotiatorFactory(registry);
+
+ final Bootstrap bootstrap = dispatcher.createClientBootStrap(Optional.<KeyMapping>absent(), Epoll.isAvailable() ? new EpollEventLoopGroup() : new NioEventLoopGroup());
+ bootstrap.localAddress(localAddress);
+ bootstrap.option(ChannelOption.SO_REUSEADDR, true);
+ bootstrap.handler(new ChannelInitializer<SocketChannel>() {
+ @Override
+ protected void initChannel(final SocketChannel ch) throws Exception {
+ ch.pipeline().addLast(hf.getDecoders());
+ ch.pipeline().addLast("negotiator", snf.getSessionNegotiator(ch, new DefaultPromise<BGPSessionImpl>(ch.eventLoop())));
+ ch.pipeline().addLast(hf.getEncoders());
+ }
+ });
+ return bootstrap.connect(remoteAddress).sync();
+ }
+
+ private static void configurePeer(final Ipv4Address localAddress, final RIBImpl ribImpl, final BgpParameters bgpParameters, final PeerRole peerRole) {
+ final InetAddress inetAddress = InetAddresses.forString(localAddress.getValue());
+
+ final BGPPeer bgpPeer = new BGPPeer(inetAddress.getHostAddress(), ribImpl, peerRole, null);
+ final List<BgpParameters> tlvs = Lists.newArrayList(bgpParameters);
+ StrictBGPPeerRegistry.GLOBAL.addPeer(new IpAddress(new Ipv4Address(inetAddress.getHostAddress())), bgpPeer,
+ new BGPSessionPreferences(AS_NUMBER, HOLDTIMER, new BgpId(RIB_ID),
+ AS_NUMBER, tlvs, Optional.<byte[]>absent()));
+ bgpPeer.instantiateServiceInstance();
+ }
+
+ private static Channel connectPeer(final Ipv4Address localAddress, final RIBImpl ribImpl, final BgpParameters bgpParameters,
+ final BGPDispatcherImpl dispatcherImpl, final BGPHandlerFactory hf, final BGPSessionListener sessionListsner) throws InterruptedException, ExecutionException {
+ final BGPPeerRegistry peerRegistry = new StrictBGPPeerRegistry();
+ peerRegistry.addPeer(new IpAddress(new Ipv4Address(RIB_ID)), sessionListsner,
+ new BGPSessionPreferences(AS_NUMBER, HOLDTIMER, new BgpId(localAddress),
+ AS_NUMBER, Lists.newArrayList(bgpParameters),Optional.<byte[]>absent()));
+
+ final ChannelFuture createClient = createClient(dispatcherImpl, new InetSocketAddress(RIB_ID, PORT), peerRegistry, new InetSocketAddress(localAddress.getValue(), PORT), hf);
+ Thread.sleep(1000);
+ return createClient.channel();
+ }
+
+ protected static BgpParameters createParameter(final boolean addPath) {
+ final OptionalCapabilities mp = new OptionalCapabilitiesBuilder().setCParameters(
+ new CParametersBuilder().addAugmentation(CParameters1.class,
+ new CParameters1Builder().setMultiprotocolCapability(
+ new MultiprotocolCapabilityBuilder().setAfi(Ipv4AddressFamily.class).setSafi(UnicastSubsequentAddressFamily.class)
+ .build()).build()).build()).build();
+ final List<OptionalCapabilities> capabilities = Lists.newArrayList(mp);
+ if (addPath) {
+ final OptionalCapabilities addPathCapa = new OptionalCapabilitiesBuilder().setCParameters(
+ new CParametersBuilder().addAugmentation(CParameters1.class,
+ new CParameters1Builder().setAddPathCapability(
+ new AddPathCapabilityBuilder().setAddressFamilies(Lists.newArrayList(
+ new AddressFamiliesBuilder()
+ .setAfi(Ipv4AddressFamily.class)
+ .setSafi(UnicastSubsequentAddressFamily.class)
+ .setSendReceive(SendReceive.Both)
+ .build()))
+ .build()).build()).build()).build();
+ capabilities.add(addPathCapa);
+ }
+ return new BgpParametersBuilder().setOptionalCapabilities(capabilities).build();
+ }
+
+ private static Update createSimpleUpdate(final Ipv4Prefix prefix, final PathId pathId, final ClusterIdentifier clusterId,
+ final long localPreference) {
+ final AttributesBuilder attBuilder = new AttributesBuilder();
+ attBuilder.setLocalPref(new LocalPrefBuilder().setPref(localPreference).build());
+ attBuilder.setOrigin(new OriginBuilder().setValue(BgpOrigin.Igp).build());
+ attBuilder.setAsPath(new AsPathBuilder().setSegments(Collections.emptyList()).build());
+ if (clusterId != null) {
+ attBuilder.setClusterId(new ClusterIdBuilder().setCluster(Collections.singletonList(clusterId)).build());
+ attBuilder.setOriginatorId(new OriginatorIdBuilder().setOriginator(new Ipv4Address(clusterId)).build());
+ }
+ attBuilder.setUnrecognizedAttributes(Collections.emptyList());
+ attBuilder.addAugmentation(Attributes1.class,
+ new Attributes1Builder().setMpReachNlri(
+ new MpReachNlriBuilder()
+ .setCNextHop(new Ipv4NextHopCaseBuilder().setIpv4NextHop(new Ipv4NextHopBuilder().setGlobal(NH1).build()).build())
+ .setAfi(Ipv4AddressFamily.class)
+ .setSafi(UnicastSubsequentAddressFamily.class)
+ .setAdvertizedRoutes(new AdvertizedRoutesBuilder().setDestinationType(
+ new DestinationIpv4CaseBuilder().setDestinationIpv4(
+ new DestinationIpv4Builder().setIpv4Prefixes(Collections.singletonList(
+ new Ipv4PrefixesBuilder().setPathId(pathId).setPrefix(new Ipv4Prefix(prefix)).build())).build())
+ .build()).build())
+ .build()).build());
+ return new UpdateBuilder().setAttributes(attBuilder.build()).build();
+ }
+}
--- /dev/null
+/*
+ * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.protocol.bgp.rib.impl;
+
+import static org.junit.Assert.assertEquals;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
+import io.netty.channel.Channel;
+import java.net.InetSocketAddress;
+import java.util.List;
+import java.util.Map;
+import org.junit.Test;
+import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
+import org.opendaylight.protocol.bgp.mode.impl.add.all.paths.AllPathSelection;
+import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Update;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParameters;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerRole;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.RibId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
+import org.opendaylight.yangtools.yang.binding.Notification;
+
+public class AddPathAllPathsTest extends AbstractAddPathTest {
+ /*
+ * All-Paths
+ * ___________________
+ * | ODL BGP 127.0.0.1 |
+ * [peer://127.0.0.2; p1, lp100] --(iBGP)--> | | --(RR-client, non add-path) --> [Peer://127.0.0.5; (p1, lp100), (p1, lp1200)]
+ * [peer://127.0.0.3; p1, lp200] --(iBGP)--> | |
+ * [peer://127.0.0.4; p1, lp50] --(iBGP)--> | | --(RR-client, add-path) --> [Peer://127.0.0.6; (p1, path-id1, lp100), (p1, path-id2, pl50), (p1, path-id3, pl200), (p1, path-id4, pl20)]
+ * [peer://127.0.0.2; p1, lp20] --(iBGP)--> |___________________|
+ * p1 = 1.1.1.1/32
+ */
+ @Test
+ public void testUseCase1() throws Exception {
+
+ final List<BgpTableType> tables = ImmutableList.of(new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class));
+ final TablesKey tk = new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
+ final Map<TablesKey, PathSelectionMode> pathTables = ImmutableMap.of(tk, new AllPathSelection());
+
+
+ final RIBImpl ribImpl = new RIBImpl(this.clusterSingletonServiceProvider, new RibId("test-rib"), AS_NUMBER, new BgpId(RIB_ID), null, this.ribExtension,
+ this.dispatcher, this.mappingService.getCodecFactory(), getDomBroker(), tables, pathTables, this.ribExtension.getClassLoadingStrategy(), null);
+
+ ribImpl.instantiateServiceInstance();
+ ribImpl.onGlobalContextUpdated(this.schemaContext);
+
+ this.dispatcher.createServer(StrictBGPPeerRegistry.GLOBAL, new InetSocketAddress(RIB_ID, PORT)).sync();
+ Thread.sleep(1000);
+
+
+ final BGPHandlerFactory hf = new BGPHandlerFactory(this.context.getMessageRegistry());
+ final BgpParameters nonAddPathParams = createParameter(false);
+ final BgpParameters addPathParams = createParameter(true);
+
+ final Channel session1 = createPeerSession(PEER1, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
+ final Channel session2 = createPeerSession(PEER2, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
+ final Channel session3 = createPeerSession(PEER3, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
+ final SimpleSessionListener listener4 = new SimpleSessionListener();
+ final Channel session4 = createPeerSession(PEER4, PeerRole.RrClient, nonAddPathParams, ribImpl, hf, listener4);
+ final SimpleSessionListener listener5 = new SimpleSessionListener();
+ final Channel session5 = createPeerSession(PEER5, PeerRole.RrClient, addPathParams, ribImpl, hf, listener5);
+ Thread.sleep(1000);
+ checkPeersPresentOnDataStore(5);
+
+ //the best route
+ sendRouteAndCheckIsOnDS(session1, PREFIX1, 100, 1);
+ assertEquals(1, listener4.getListMsg().size());
+ assertEquals(1, listener5.getListMsg().size());
+ assertEquals(UPD_100, listener5.getListMsg().get(0));
+
+ //the second best route
+ sendRouteAndCheckIsOnDS(session2, PREFIX1, 50, 2);
+ assertEquals(1, listener4.getListMsg().size());
+ assertEquals(2, listener5.getListMsg().size());
+ assertEquals(UPD_50, listener5.getListMsg().get(1));
+
+ //new best route
+ sendRouteAndCheckIsOnDS(session3, PREFIX1, 200, 3);
+ assertEquals(2, listener4.getListMsg().size());
+ assertEquals(3, listener5.getListMsg().size());
+ assertEquals(UPD_200, listener5.getListMsg().get(2));
+
+ //the worst route
+ sendRouteAndCheckIsOnDS(session1, PREFIX1, 20, 3);
+ assertEquals(2, listener4.getListMsg().size());
+ assertEquals(4, listener5.getListMsg().size());
+ assertEquals(UPD_200.getAttributes().getLocalPref(), ((Update) listener4.getListMsg().get(1)).getAttributes().getLocalPref());
+ assertEquals(UPD_20, listener5.getListMsg().get(3));
+
+ session1.close();
+ session2.close();
+ session3.close();
+ session4.close();
+ session5.close();
+ }
+}
--- /dev/null
+/*
+ * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.protocol.bgp.rib.impl;
+
+import static org.junit.Assert.assertEquals;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
+import io.netty.channel.Channel;
+import java.net.InetSocketAddress;
+import java.util.List;
+import java.util.Map;
+import org.junit.Test;
+import org.opendaylight.protocol.bgp.mode.api.PathSelectionMode;
+import org.opendaylight.protocol.bgp.mode.impl.add.n.paths.AddPathBestNPathSelection;
+import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParameters;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerRole;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.RibId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.BgpId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
+
+public class AddPathNPathsTest extends AbstractAddPathTest {
+
+ /*
+ * N-Paths
+ * ___________________
+ * | ODL BGP 127.0.0.1 |
+ * [peer://127.0.0.2; p1, lp100] --(iBGP)--> | | --(RR-client, non add-path) --> [Peer://127.0.0.5; (p1, lp100), (p1, lp1200)]
+ * [peer://127.0.0.3; p1, lp200] --(iBGP)--> | |
+ * [peer://127.0.0.4; p1, lp50] --(iBGP)--> | | --(RR-client, add-path) --> [Peer://127.0.0.6; (p1, path-id1, lp100), (p1, path-id2, pl50), (p1, path-id3, pl200)]
+ * [peer://127.0.0.2; p1, lp20] --(iBGP)--> |___________________|
+ * p1 = 1.1.1.1/32
+ */
+ @Test
+ public void testUseCase1() throws Exception {
+
+ final List<BgpTableType> tables = ImmutableList.of(new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class));
+ final TablesKey tk = new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
+ final Map<TablesKey, PathSelectionMode> pathTables = ImmutableMap.of(tk, new AddPathBestNPathSelection(2L));
+
+ final RIBImpl ribImpl = new RIBImpl(this.clusterSingletonServiceProvider, new RibId("test-rib"), AS_NUMBER, new BgpId(RIB_ID), null, this.ribExtension,
+ this.dispatcher, this.mappingService.getCodecFactory(), getDomBroker(), tables, pathTables, this.ribExtension.getClassLoadingStrategy(), null);
+
+ ribImpl.instantiateServiceInstance();
+ ribImpl.onGlobalContextUpdated(this.schemaContext);
+
+ this.dispatcher.createServer(StrictBGPPeerRegistry.GLOBAL, new InetSocketAddress(RIB_ID, PORT)).sync();
+ Thread.sleep(1000);
+
+ final BGPHandlerFactory hf = new BGPHandlerFactory(this.context.getMessageRegistry());
+ final BgpParameters nonAddPathParams = createParameter(false);
+ final BgpParameters addPathParams = createParameter(true);
+
+ final Channel session1 = createPeerSession(PEER1, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
+ final Channel session2 = createPeerSession(PEER2, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
+ final Channel session3 = createPeerSession(PEER3, PeerRole.Ibgp, nonAddPathParams, ribImpl, hf, new SimpleSessionListener());
+ final SimpleSessionListener listener4 = new SimpleSessionListener();
+ final Channel session4 = createPeerSession(PEER4, PeerRole.RrClient, nonAddPathParams, ribImpl, hf, listener4);
+ final SimpleSessionListener listener5 = new SimpleSessionListener();
+ final Channel session5 = createPeerSession(PEER5, PeerRole.RrClient, addPathParams, ribImpl, hf, listener5);
+ Thread.sleep(1000);
+ checkPeersPresentOnDataStore(5);
+
+ //new best route so far
+ sendRouteAndCheckIsOnDS(session1, PREFIX1, 100, 1);
+ assertEquals(1, listener4.getListMsg().size());
+ assertEquals(1, listener5.getListMsg().size());
+ assertEquals(UPD_100, listener5.getListMsg().get(0));
+
+ //the second best route
+ sendRouteAndCheckIsOnDS(session2, PREFIX1, 50, 2);
+ assertEquals(1, listener4.getListMsg().size());
+ assertEquals(2, listener5.getListMsg().size());
+ assertEquals(UPD_50, listener5.getListMsg().get(1));
+
+ //new best route
+ sendRouteAndCheckIsOnDS(session3, PREFIX1, 200, 2);
+ assertEquals(2, listener4.getListMsg().size());
+ assertEquals(3, listener5.getListMsg().size());
+ assertEquals(UPD_200, listener5.getListMsg().get(2));
+
+ //the worst prefix, no changes
+ sendRouteAndCheckIsOnDS(session2, PREFIX1, 20, 2);
+ assertEquals(2, listener4.getListMsg().size());
+ assertEquals(3, listener5.getListMsg().size());
+
+ session1.close();
+ session2.close();
+ session3.close();
+ session4.close();
+ session5.close();
+ }
+}