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.parser.impl.BGPActivator;
  23 import org.opendaylight.protocol.bgp.parser.spi.BGPExtensionProviderContext;
  24 import org.opendaylight.protocol.bgp.parser.spi.pojo.ServiceLoaderBGPExtensionProviderContext;
  25 import org.opendaylight.protocol.bgp.rib.impl.BGPDispatcherImpl;
  26 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPDispatcher;
  27 import org.opendaylight.protocol.bgp.rib.spi.BGPSessionListener;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.AsNumber;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.LinkstateAddressFamily;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.LinkstateSubsequentAddressFamily;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParameters;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.BgpParametersBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.OptionalCapabilities;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.OptionalCapabilitiesBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.message.bgp.parameters.optional.capabilities.c.parameters.As4BytesCapabilityBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.CParameters1Builder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.SendReceive;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.AddPathCapabilityBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.MultiprotocolCapabilityBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.mp.capabilities.add.path.capability.AddressFamiliesBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.AddressFamily;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.SubsequentAddressFamily;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
  47
  48 final class BGPTestTool {
  49     private final Map<String, BGPSessionListener> listeners = new HashMap<>();
  50
  51     void start(final Arguments arguments) {
  52         final BGPDispatcher dispatcher = initializeActivator();
  53
  54         final ArrayList<OptionalCapabilities> optCap = Lists.newArrayList(createMPCapability(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class),
  55             createMPCapability(LinkstateAddressFamily.class, LinkstateSubsequentAddressFamily.class), createAs4BytesMPCapability(arguments.getAs()));
  56         if (arguments.getMultiPathSupport()) {
  57             optCap.add(createAddPathCapability());
  58         }
  59         final BgpParameters bgpParameters = createBgpParameters(optCap);
  60
  61         final InetSocketAddress localAddress = arguments.getLocalAddresses();
  62         final int port = localAddress.getPort();
  63         InetAddress address = localAddress.getAddress();
  64         int numberOfSpeakers = arguments.getSpeakerCount();
  65         do {
  66             final BGPSessionListener sessionListener = new TestingListener(arguments.getNumberOfPrefixes(), arguments.getExtendedCommunities(),
  67                 arguments.getMultiPathSupport());
  68             this.listeners.put(address.getHostAddress(), sessionListener);
  69             createPeer(dispatcher, arguments, new InetSocketAddress(address, port), sessionListener, bgpParameters);
  70             numberOfSpeakers--;
  71             address = InetAddresses.increment(address);
  72         } while (numberOfSpeakers > 0);
  73     }
  74
  75     private BGPDispatcher initializeActivator() {
  76         final BGPActivator activator = new BGPActivator();
  77         final BGPExtensionProviderContext ctx = ServiceLoaderBGPExtensionProviderContext.getSingletonInstance();
  78         activator.start(ctx);
  79         return new BGPDispatcherImpl(ctx.getMessageRegistry(), new NioEventLoopGroup(), new NioEventLoopGroup());
  80     }
  81
  82     private static OptionalCapabilities createMPCapability(final Class<? extends AddressFamily> afi, final Class<? extends SubsequentAddressFamily> safi) {
  83         return new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder().addAugmentation(CParameters1.class, new CParameters1Builder()
  84             .setMultiprotocolCapability(new MultiprotocolCapabilityBuilder().setAfi(afi).setSafi(safi).build()).build()).build()).build();
  85     }
  86
  87     private static OptionalCapabilities createAs4BytesMPCapability(final AsNumber as) {
  88         return new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder()
  89             .setAs4BytesCapability(new As4BytesCapabilityBuilder().setAsNumber(as).build()).build()).build();
  90     }
  91
  92     private static BgpParameters createBgpParameters(final List<OptionalCapabilities> optionalCapabilities) {
  93         return new BgpParametersBuilder().setOptionalCapabilities(optionalCapabilities).build();
  94     }
  95
  96     private static OptionalCapabilities createAddPathCapability() {
  97         return new OptionalCapabilitiesBuilder().setCParameters(new CParametersBuilder().addAugmentation(CParameters1.class, new CParameters1Builder()
  98             .setAddPathCapability(new AddPathCapabilityBuilder().setAddressFamilies(Lists.newArrayList(new AddressFamiliesBuilder()
  99                 .setAfi(Ipv4AddressFamily.class).setSafi(UnicastSubsequentAddressFamily.class).setSendReceive(SendReceive.Both).build()
 100             )).build()).build()).build()).build();
 101     }
 102
 103     void printCount(final String localAddress) {
 104         final BGPSessionListener listener = this.listeners.get(localAddress);
 105         if (listener != null) {
 106             ((TestingListener) listener).printCount(localAddress);
 107         }
 108     }
 109 }