bgp/testtool/src/main/java/org/opendaylight/protocol/bgp/testtool/BGPTestTool.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
   9 package org.opendaylight.protocol.bgp.testtool;
  10
  11 import static org.opendaylight.protocol.bgp.testtool.BGPPeerBuilder.createPeer;
  12
  13 import com.google.common.collect.Lists;
  14 import com.google.common.net.InetAddresses;
  15 import io.netty.channel.nio.NioEventLoopGroup;
  16 import java.net.InetAddress;
  17 import java.net.InetSocketAddress;
  18 import java.util.ArrayList;
  19 import java.util.HashMap;
  20 import java.util.List;
  21 import java.util.Map;
  22 import org.opendaylight.protocol.bgp.flowspec.FlowspecActivator;
  23 import org.opendaylight.protocol.bgp.flowspec.SimpleFlowspecExtensionProviderContext;
  24 import org.opendaylight.protocol.bgp.parser.impl.BGPActivator;
  25 import org.opendaylight.protocol.bgp.parser.spi.BGPExtensionProviderContext;
  26 import org.opendaylight.protocol.bgp.parser.spi.pojo.ServiceLoaderBGPExtensionProviderContext;
  27 import org.opendaylight.protocol.bgp.rib.impl.BGPDispatcherImpl;
  28 import org.opendaylight.protocol.bgp.rib.impl.StrictBGPPeerRegistry;
  29 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPDispatcher;
  30 import org.opendaylight.protocol.bgp.rib.spi.BGPSessionListener;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.EvpnSubsequentAddressFamily;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.L2vpnAddressFamily;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.l3vpn.mcast.rev180417.McastMplsLabeledVpnSubsequentAddressFamily;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev200120.LinkstateAddressFamily;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev200120.LinkstateSubsequentAddressFamily;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.BgpParameters;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.BgpParametersBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.bgp.parameters.OptionalCapabilities;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.bgp.parameters.OptionalCapabilitiesBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.open.message.bgp.parameters.optional.capabilities.c.parameters.As4BytesCapabilityBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.CParameters1;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.CParameters1Builder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.SendReceive;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.AddPathCapabilityBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.MultiprotocolCapabilityBuilder;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.add.path.capability.AddressFamiliesBuilder;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.RouteTargetConstrainSubsequentAddressFamily;
  50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.AddressFamily;
  51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.Ipv4AddressFamily;
  52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.Ipv6AddressFamily;
  53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.MplsLabeledVpnSubsequentAddressFamily;
  54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.SubsequentAddressFamily;
  55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.UnicastSubsequentAddressFamily;
  56
  57 final class BGPTestTool {
  58     private final Map<String, BGPSessionListener> listeners = new HashMap<>();
  59
  60     void start(final Arguments arguments) {
  61         final BGPDispatcher dispatcher = initializeActivator();
  62
  63         final ArrayList<OptionalCapabilities> optCap = Lists.newArrayList(createMPCapability(Ipv4AddressFamily.class,
  64                 UnicastSubsequentAddressFamily.class),
  65             createMPCapability(LinkstateAddressFamily.class, LinkstateSubsequentAddressFamily.class),
  66             createMPCapability(Ipv4AddressFamily.class, MplsLabeledVpnSubsequentAddressFamily.class),
  67             createMPCapability(Ipv6AddressFamily.class, MplsLabeledVpnSubsequentAddressFamily.class),
  68             createMPCapability(Ipv4AddressFamily.class, McastMplsLabeledVpnSubsequentAddressFamily.class),
  69             createMPCapability(Ipv6AddressFamily.class, McastMplsLabeledVpnSubsequentAddressFamily.class),
  70             createMPCapability(L2vpnAddressFamily.class, EvpnSubsequentAddressFamily.class),
  71             createMPCapability(Ipv4AddressFamily.class, RouteTargetConstrainSubsequentAddressFamily.class),
  72                 createAs4BytesMPCapability(arguments.getAs()));
  73         if (arguments.getMultiPathSupport()) {
  74             optCap.add(createAddPathCapability());
  75         }
  76         final BgpParameters bgpParameters = createBgpParameters(optCap);
  77
  78         final InetSocketAddress localAddress = arguments.getLocalAddresses();
  79         final int port = localAddress.getPort();
  80         InetAddress address = localAddress.getAddress();
  81         int numberOfSpeakers = arguments.getSpeakerCount();
  82         do {
  83             final BGPSessionListener sessionListener = new TestingListener(arguments.getNumberOfPrefixes(),
  84                     arguments.getExtendedCommunities(),
  85                 arguments.getMultiPathSupport());
  86             this.listeners.put(address.getHostAddress(), sessionListener);
  87             createPeer(dispatcher, arguments, new InetSocketAddress(address, port), sessionListener, bgpParameters);
  88             numberOfSpeakers--;
  89             address = InetAddresses.increment(address);
  90         } while (numberOfSpeakers > 0);
  91     }
  92
  93     private static BGPDispatcher initializeActivator() {
  94         final BGPActivator activator = new BGPActivator();
  95         final BGPExtensionProviderContext ctx = ServiceLoaderBGPExtensionProviderContext.getSingletonInstance();
  96         activator.start(ctx);
  97
  98         final org.opendaylight.protocol.bgp.inet.BGPActivator inetActivator
  99                 = new org.opendaylight.protocol.bgp.inet.BGPActivator();
 100         inetActivator.start(ctx);
 101
 102         final org.opendaylight.protocol.bgp.evpn.impl.BGPActivator evpnActivator = new org.opendaylight.protocol.bgp
 103                 .evpn.impl.BGPActivator();
 104         evpnActivator.start(ctx);
 105
 106         final SimpleFlowspecExtensionProviderContext fsContext = new SimpleFlowspecExtensionProviderContext();
 107         final FlowspecActivator flowspecActivator = new FlowspecActivator(fsContext);
 108         final org.opendaylight.protocol.bgp.flowspec.BGPActivator flowspecBGPActivator
 109                 = new org.opendaylight.protocol.bgp.flowspec.BGPActivator(flowspecActivator);
 110         flowspecBGPActivator.start(ctx);
 111
 112         final org.opendaylight.protocol.bgp.labeled.unicast.BGPActivator labeledActivator
 113                 = new org.opendaylight.protocol.bgp.labeled.unicast.BGPActivator();
 114         labeledActivator.start(ctx);
 115
 116         final org.opendaylight.protocol.bgp.l3vpn.BGPActivator l3vpnBGPActivator
 117                 = new org.opendaylight.protocol.bgp.l3vpn.BGPActivator();
 118         l3vpnBGPActivator.start(ctx);
 119
 120         final org.opendaylight.protocol.bgp.route.targetcontrain.impl.activators.BGPActivator rtBGPActivator
 121                 = new org.opendaylight.protocol.bgp.route.targetcontrain.impl.activators.BGPActivator();
 122         rtBGPActivator.start(ctx);
 123
 124         return new BGPDispatcherImpl(ctx.getMessageRegistry(), new NioEventLoopGroup(), new NioEventLoopGroup(),
 125             new StrictBGPPeerRegistry());
 126     }
 127
 128     private static OptionalCapabilities createMPCapability(final Class<? extends AddressFamily> afi,
 129             final Class<? extends SubsequentAddressFamily> safi) {
 130         return new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder()
 131                 .addAugmentation(CParameters1.class, new CParameters1Builder()
 132             .setMultiprotocolCapability(new MultiprotocolCapabilityBuilder().setAfi(afi)
 133                     .setSafi(safi).build()).build()).build()).build();
 134     }
 135
 136     private static OptionalCapabilities createAs4BytesMPCapability(final AsNumber as) {
 137         return new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder()
 138             .setAs4BytesCapability(new As4BytesCapabilityBuilder().setAsNumber(as).build()).build()).build();
 139     }
 140
 141     private static BgpParameters createBgpParameters(final List<OptionalCapabilities> optionalCapabilities) {
 142         return new BgpParametersBuilder().setOptionalCapabilities(optionalCapabilities).build();
 143     }
 144
 145     private static OptionalCapabilities createAddPathCapability() {
 146         return new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder()
 147                 .addAugmentation(CParameters1.class, new CParameters1Builder()
 148                         .setAddPathCapability(new AddPathCapabilityBuilder()
 149                                 .setAddressFamilies(Lists.newArrayList(new AddressFamiliesBuilder()
 150                                         .setAfi(Ipv4AddressFamily.class)
 151                                         .setSafi(UnicastSubsequentAddressFamily.class)
 152                                         .setSendReceive(SendReceive.Both).build())).build()).build()).build()).build();
 153     }
 154
 155     void printCount(final String localAddress) {
 156         final BGPSessionListener listener = this.listeners.get(localAddress);
 157         if (listener != null) {
 158             ((TestingListener) listener).printCount(localAddress);
 159         }
 160     }
 161 }