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