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.linkstate.rev180329.LinkstateAddressFamily;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev180329.LinkstateSubsequentAddressFamily;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.BgpParameters;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.BgpParametersBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.bgp.parameters.OptionalCapabilities;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.bgp.parameters.OptionalCapabilitiesBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.open.message.bgp.parameters.optional.capabilities.c.parameters.As4BytesCapabilityBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.CParameters1;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.CParameters1Builder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.SendReceive;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.mp.capabilities.AddPathCapabilityBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.mp.capabilities.MultiprotocolCapabilityBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev171207.mp.capabilities.add.path.capability.AddressFamiliesBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.AddressFamily;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.SubsequentAddressFamily;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
  50
  51 final class BGPTestTool {
  52     private final Map<String, BGPSessionListener> listeners = new HashMap<>();
  53
  54     void start(final Arguments arguments) {
  55         final BGPDispatcher dispatcher = initializeActivator();
  56
  57         final ArrayList<OptionalCapabilities> optCap = Lists.newArrayList(createMPCapability(Ipv4AddressFamily.class,
  58                 UnicastSubsequentAddressFamily.class),
  59             createMPCapability(LinkstateAddressFamily.class, LinkstateSubsequentAddressFamily.class),
  60                 createAs4BytesMPCapability(arguments.getAs()));
  61         if (arguments.getMultiPathSupport()) {
  62             optCap.add(createAddPathCapability());
  63         }
  64         final BgpParameters bgpParameters = createBgpParameters(optCap);
  65
  66         final InetSocketAddress localAddress = arguments.getLocalAddresses();
  67         final int port = localAddress.getPort();
  68         InetAddress address = localAddress.getAddress();
  69         int numberOfSpeakers = arguments.getSpeakerCount();
  70         do {
  71             final BGPSessionListener sessionListener = new TestingListener(arguments.getNumberOfPrefixes(),
  72                     arguments.getExtendedCommunities(),
  73                 arguments.getMultiPathSupport());
  74             this.listeners.put(address.getHostAddress(), sessionListener);
  75             createPeer(dispatcher, arguments, new InetSocketAddress(address, port), sessionListener, bgpParameters);
  76             numberOfSpeakers--;
  77             address = InetAddresses.increment(address);
  78         } while (numberOfSpeakers > 0);
  79     }
  80
  81     private static BGPDispatcher initializeActivator() {
  82         final BGPActivator activator = new BGPActivator();
  83         final BGPExtensionProviderContext ctx = ServiceLoaderBGPExtensionProviderContext.getSingletonInstance();
  84         activator.start(ctx);
  85
  86         final org.opendaylight.protocol.bgp.inet.BGPActivator inetActivator
  87                 = new org.opendaylight.protocol.bgp.inet.BGPActivator();
  88         inetActivator.start(ctx);
  89
  90         final org.opendaylight.protocol.bgp.evpn.impl.BGPActivator evpnActivator = new org.opendaylight.protocol.bgp
  91                 .evpn.impl.BGPActivator();
  92         evpnActivator.start(ctx);
  93
  94         final SimpleFlowspecExtensionProviderContext fsContext = new SimpleFlowspecExtensionProviderContext();
  95         final FlowspecActivator flowspecActivator = new FlowspecActivator(fsContext);
  96         final org.opendaylight.protocol.bgp.flowspec.BGPActivator flowspecBGPActivator
  97                 = new org.opendaylight.protocol.bgp.flowspec.BGPActivator(flowspecActivator);
  98         flowspecBGPActivator.start(ctx);
  99
 100         final org.opendaylight.protocol.bgp.labeled.unicast.BGPActivator labeledActivator
 101                 = new org.opendaylight.protocol.bgp.labeled.unicast.BGPActivator();
 102         labeledActivator.start(ctx);
 103
 104         final org.opendaylight.protocol.bgp.l3vpn.ipv4.BgpIpv4Activator bgpIpv4Activator
 105                 = new org.opendaylight.protocol.bgp.l3vpn.ipv4.BgpIpv4Activator();
 106         bgpIpv4Activator.start(ctx);
 107
 108         final org.opendaylight.protocol.bgp.l3vpn.ipv6.BgpIpv6Activator bgpIpv6Activator
 109                 = new org.opendaylight.protocol.bgp.l3vpn.ipv6.BgpIpv6Activator();
 110         bgpIpv6Activator.start(ctx);
 111
 112         return new BGPDispatcherImpl(ctx.getMessageRegistry(), new NioEventLoopGroup(), new NioEventLoopGroup(),
 113             new StrictBGPPeerRegistry());
 114     }
 115
 116     private static OptionalCapabilities createMPCapability(final Class<? extends AddressFamily> afi,
 117             final Class<? extends SubsequentAddressFamily> safi) {
 118         return new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder()
 119                 .addAugmentation(CParameters1.class, new CParameters1Builder()
 120             .setMultiprotocolCapability(new MultiprotocolCapabilityBuilder().setAfi(afi)
 121                     .setSafi(safi).build()).build()).build()).build();
 122     }
 123
 124     private static OptionalCapabilities createAs4BytesMPCapability(final AsNumber as) {
 125         return new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder()
 126             .setAs4BytesCapability(new As4BytesCapabilityBuilder().setAsNumber(as).build()).build()).build();
 127     }
 128
 129     private static BgpParameters createBgpParameters(final List<OptionalCapabilities> optionalCapabilities) {
 130         return new BgpParametersBuilder().setOptionalCapabilities(optionalCapabilities).build();
 131     }
 132
 133     private static OptionalCapabilities createAddPathCapability() {
 134         return new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder()
 135                 .addAugmentation(CParameters1.class, new CParameters1Builder()
 136                         .setAddPathCapability(new AddPathCapabilityBuilder()
 137                                 .setAddressFamilies(Lists.newArrayList(new AddressFamiliesBuilder()
 138                                         .setAfi(Ipv4AddressFamily.class)
 139                                         .setSafi(UnicastSubsequentAddressFamily.class)
 140                                         .setSendReceive(SendReceive.Both).build())).build()).build()).build()).build();
 141     }
 142
 143     void printCount(final String localAddress) {
 144         final BGPSessionListener listener = this.listeners.get(localAddress);
 145         if (listener != null) {
 146             ((TestingListener) listener).printCount(localAddress);
 147         }
 148     }
 149 }