bmp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMockUtil.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 package org.opendaylight.protocol.bmp.mock;
   9
  10 import java.net.InetAddress;
  11 import java.util.Collections;
  12 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
  13 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  14 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  15 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
  16 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.OpenBuilder;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.OpenMessage;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.ProtocolVersion;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.AttributesBuilder;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.AsPath;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.AsPathBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.Origin;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.OriginBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.update.message.NlriBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.BgpOrigin;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.next.hop.c.next.hop.Ipv4NextHopCaseBuilder;
  28 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;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.AdjRibInType;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.InitiationMessage;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.InitiationMessageBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.PeerType;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.PeerUpNotification;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.PeerUpNotificationBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.RouteMonitoringMessage;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.RouteMonitoringMessageBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.description.tlv.DescriptionTlvBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.initiation.TlvsBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.name.tlv.NameTlvBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.peer.header.PeerHeader;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.peer.header.PeerHeaderBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.peer.up.ReceivedOpenBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.peer.up.SentOpenBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.route.monitoring.message.Update;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.route.monitoring.message.UpdateBuilder;
  46 import org.opendaylight.yangtools.yang.common.Uint16;
  47 import org.opendaylight.yangtools.yang.common.Uint32;
  48 import org.opendaylight.yangtools.yang.common.Uint8;
  49
  50 final class BmpMockUtil {
  51     private static final String DESCRIPTION = "OpenDaylight";
  52     private static final String NAME = "BMP mock";
  53     private static final int HOLD_TIMER = 180;
  54     private static final AsNumber ASN = new AsNumber(Uint32.valueOf(65431));
  55     private static final Ipv4Address NEXT_HOP = new Ipv4Address("1.2.3.4");
  56     private static final PortNumber PEER_PORT = new PortNumber(Uint16.valueOf(179).intern());
  57     private static final ProtocolVersion PROTOCOL_VERSION = new ProtocolVersion(Uint8.valueOf(4));
  58     private static final Origin ORIGIN = new OriginBuilder().setValue(BgpOrigin.Igp).build();
  59     private static final AsPath AS_PATH = new AsPathBuilder().setSegments(Collections.emptyList()).build();
  60
  61     private BmpMockUtil() {
  62         // Hidden on purpose
  63     }
  64
  65     static InitiationMessage createInitiation() {
  66         final InitiationMessageBuilder msgBuilder = new InitiationMessageBuilder();
  67         msgBuilder.setTlvs(
  68             new TlvsBuilder()
  69                 .setDescriptionTlv(new DescriptionTlvBuilder().setDescription(DESCRIPTION).build())
  70                 .setNameTlv(new NameTlvBuilder().setName(NAME).build())
  71                 .build());
  72         return msgBuilder.build();
  73     }
  74
  75     static PeerUpNotification createPeerUp(final Ipv4Address peerIp, final InetAddress localAddress) {
  76         final PeerUpNotificationBuilder msgBuilder = new PeerUpNotificationBuilder();
  77         msgBuilder.setLocalAddress(new IpAddress(new Ipv4Address(localAddress.getHostAddress())));
  78         msgBuilder.setLocalPort(PEER_PORT);
  79         msgBuilder.setRemotePort(PEER_PORT);
  80         msgBuilder.setReceivedOpen(new ReceivedOpenBuilder(createOpen(peerIp)).build());
  81         msgBuilder.setSentOpen(new SentOpenBuilder(createOpen(new Ipv4Address(localAddress.getHostAddress()))).build());
  82         msgBuilder.setPeerHeader(createPeerHeader(peerIp, AdjRibInType.PrePolicy));
  83         return msgBuilder.build();
  84     }
  85
  86     private static OpenMessage createOpen(final Ipv4Address address) {
  87         final OpenBuilder msgBuilder = new OpenBuilder();
  88         msgBuilder.setBgpIdentifier(address);
  89         msgBuilder.setHoldTimer(HOLD_TIMER);
  90         msgBuilder.setMyAsNumber(Uint16.valueOf(ASN.getValue()));
  91         msgBuilder.setVersion(PROTOCOL_VERSION);
  92         return msgBuilder.build();
  93     }
  94
  95     private static PeerHeader createPeerHeader(final Ipv4Address bgpId, final AdjRibInType ribType) {
  96         return new PeerHeaderBuilder()
  97             .setAddress(new IpAddress(bgpId))
  98             .setAdjRibInType(ribType)
  99             .setAs(new AsNumber(ASN))
 100             .setBgpId(bgpId)
 101             .setIpv4(true)
 102             .setType(PeerType.Global)
 103             .build();
 104     }
 105
 106     static RouteMonitoringMessage createRouteMonitoring(final Ipv4Address bgpId, final AdjRibInType ribType,
 107             final Ipv4Prefix prefix) {
 108         final RouteMonitoringMessageBuilder routeMonitMsgBuilder = new RouteMonitoringMessageBuilder()
 109             .setPeerHeader(createPeerHeader(bgpId, ribType))
 110             .setUpdate(createUpdate(prefix));
 111         return routeMonitMsgBuilder.build();
 112     }
 113
 114     private static Update createUpdate(final Ipv4Prefix prefix) {
 115         final UpdateBuilder updateBuilder = new UpdateBuilder()
 116             .setAttributes(new AttributesBuilder().setOrigin(ORIGIN).setAsPath(AS_PATH).setCNextHop(
 117                 new Ipv4NextHopCaseBuilder().setIpv4NextHop(new Ipv4NextHopBuilder().setGlobal(NEXT_HOP).build())
 118                         .build()).build())
 119             .setNlri(Collections.singletonList(new NlriBuilder().setPrefix(prefix).build()));
 120         return updateBuilder.build();
 121     }
 122 }