Skip to content

Commit

Permalink
Generate network map from BGP local rib
Browse files Browse the repository at this point in the history 
Signed-off-by: jingxuanzhang <[email protected]>
  • Loading branch information
@fno2010
fno2010 committed Feb 19, 2020
1 parent a1c9e58 commit f538621
Show file tree
Hide file tree
Showing 30 changed files with 626 additions and 154 deletions.
15 changes: 15 additions & 0 deletions alto-basic/auto-maps/api/pom.xml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -21,4 +21,19 @@ and is available at http://www.eclipse.org/legal/epl-v10.html
<version>0.6.4</version>
<name>ODL :: alto :: ${project.artifactId}</name>
<packaging>bundle</packaging>
<dependencies>
<dependency>
<groupId>org.opendaylight.alto.core</groupId>
<artifactId>alto-basic-types</artifactId>
<version>${project.version}</version>
</dependency>
<dependency>
<groupId>org.opendaylight.mdsal.model</groupId>
<artifactId>ietf-yang-types-20130715</artifactId>
</dependency>
<dependency>
<groupId>org.opendaylight.mdsal.model</groupId>
<artifactId>ietf-topology</artifactId>
</dependency>
</dependencies>
</project>
74 changes: 74 additions & 0 deletions alto-basic/auto-maps/api/src/main/yang/alto-auto-maps.yang
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,81 @@ module alto-auto-maps {
namespace "urn:alto:auto:maps";
prefix "alto-auto-maps";

import "ietf-yang-types" {
prefix "ietf-types";
}

import "network-topology" {
prefix "topology";
}

import "alto-basic-types" {
prefix "alto-types";
}

organization "Yale University";

contact "[email protected]";

revision "2015-01-05" {
description "Initial revision of alto-auto-maps model";
}

typedef topology-type {
type enumeration {
enum openflow;
enum bgp-ipv4;
enum bgp-ls;
}
}

// TODO: Augment to config-context of manual map
list config-context {
key "context-id";

leaf "context-id" {
type ietf-types:uuid;
default "00000000-0000-0000-0000-000000000000";
}

list resource-network-map {
key "resource-id";

leaf "resource-id" {
mandatory true;
type alto-types:resource-id;
}

leaf "topology-id" {
type string;
default "flow:1";
}

leaf "topology-type" {
type topology-type;
default openflow;
}

choice algorithm {
case spt-cluster {
container spt-cluster-params {
leaf spt-root {
type topology:node-id;
}
leaf num-of-clusters {
type int32;
}
}
}
case bgp-simple-as-cluster {
container bgp-simple-as-params {
// FIXME: identityref is better
leaf bgp-rib {
type string;
}
}
}
}
}
}
}
13 changes: 13 additions & 0 deletions alto-basic/auto-maps/impl/pom.xml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ and is available at http://www.eclipse.org/legal/epl-v10.html

<properties>
<l2switch.version>0.7.4</l2switch.version>
<bgpcep.version>0.9.4</bgpcep.version>
</properties>

<dependencies>
Expand Down Expand Up @@ -69,6 +70,18 @@ and is available at http://www.eclipse.org/legal/epl-v10.html
<version>${l2switch.version}</version>
</dependency>

<dependency>
<groupId>org.opendaylight.bgpcep</groupId>
<artifactId>bgp-rib-api</artifactId>
<version>${bgpcep.version}</version>
</dependency>

<dependency>
<groupId>org.opendaylight.bgpcep</groupId>
<artifactId>bgp-inet</artifactId>
<version>${bgpcep.version}</version>
</dependency>

<!-- Testing Dependencies -->
<dependency>
<groupId>junit</groupId>
Expand Down
151 changes: 151 additions & 0 deletions ...-maps/impl/src/main/java/org/opendaylight/alto/basic/impl/AltoAutoMapsBgpIpv4Updater.java
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,151 @@
/*
* Copyright © 2015 Yale University and others. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
package org.opendaylight.alto.basic.impl;

import com.google.common.base.Optional;
import org.opendaylight.alto.basic.manual.maps.ManualMapsUtils;
import org.opendaylight.controller.md.sal.binding.api.DataBroker;
import org.opendaylight.controller.md.sal.binding.api.ReadTransaction;
import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
import org.opendaylight.yang.gen.v1.urn.alto.auto.maps.rev150105.config.context.ResourceNetworkMap;
import org.opendaylight.yang.gen.v1.urn.alto.auto.maps.rev150105.config.context.resource.network.map.algorithm.BgpSimpleAsCluster;
import org.opendaylight.yang.gen.v1.urn.alto.manual.maps.networkmap.rev151021.EndpointAddressType;
import org.opendaylight.yang.gen.v1.urn.alto.manual.maps.networkmap.rev151021.endpoint.address.group.EndpointAddressGroup;
import org.opendaylight.yang.gen.v1.urn.alto.manual.maps.networkmap.rev151021.endpoint.address.group.EndpointAddressGroupBuilder;
import org.opendaylight.yang.gen.v1.urn.alto.manual.maps.networkmap.rev151021.network.map.Map;
import org.opendaylight.yang.gen.v1.urn.alto.manual.maps.networkmap.rev151021.network.map.MapBuilder;
import org.opendaylight.yang.gen.v1.urn.alto.types.rev150921.PidName;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpPrefix;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.bgp.rib.rib.loc.rib.tables.routes.Ipv4RoutesCase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev171207.ipv4.routes.ipv4.routes.Ipv4Route;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.as.path.Segments;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.BgpRib;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.RibId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.bgp.rib.Rib;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.bgp.rib.RibKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.bgp.rib.rib.LocRib;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.rib.Tables;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev171207.rib.TablesKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Arrays;
import java.util.LinkedHashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.concurrent.ExecutionException;

public class AltoAutoMapsBgpIpv4Updater implements AutoCloseable {

private static final Logger LOG = LoggerFactory.getLogger(AltoAutoMapsBgpIpv4Updater.class);

private final DataBroker dataBroker;
// private ListenerRegistration<?> registration;
private String contextId;
private ResourceNetworkMap networkMapConfig;

public AltoAutoMapsBgpIpv4Updater(String contextId, ResourceNetworkMap networkMapConfig, final DataBroker dataBroker) {
this.dataBroker = dataBroker;
this.contextId = contextId;
this.networkMapConfig = networkMapConfig;
registerBGPListener();
}

private void registerBGPListener() {
final ReadWriteTransaction wrx = dataBroker.newReadWriteTransaction();
try {
List<Map> networkMap = computeNetworkMapByBgpIpv4(networkMapConfig);
LOG.info("Putting auto generated network-map to manual map config...");
ManualMapsUtils.createResourceNetworkMap(contextId, networkMapConfig.getResourceId().getValue(),
networkMap, wrx);
} catch (InterruptedException e) {
e.printStackTrace();
} catch (ExecutionException e) {
e.printStackTrace();
}
wrx.submit();
}

private List<Map> computeNetworkMapByBgpIpv4(ResourceNetworkMap networkMapConfig) throws InterruptedException, ExecutionException {
final ReadTransaction rx = dataBroker.newReadOnlyTransaction();
List<Map> networkMap = new LinkedList<>();
if (networkMapConfig.getAlgorithm() instanceof BgpSimpleAsCluster) {
BgpSimpleAsCluster algorithm = (BgpSimpleAsCluster) networkMapConfig.getAlgorithm();
String ribId = algorithm.getBgpSimpleAsParams().getBgpRib();
InstanceIdentifier<Tables> ribIID = InstanceIdentifier.builder(BgpRib.class)
.child(Rib.class, new RibKey(new RibId(ribId)))
.child(LocRib.class)
.child(Tables.class, new TablesKey(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class))
.build();
try {
Optional<Tables> optional = rx.read(LogicalDatastoreType.OPERATIONAL, ribIID).get();
if (optional.isPresent()) {
Tables table = optional.get();
if (table.getRoutes() instanceof Ipv4RoutesCase) {
Ipv4RoutesCase routesCase = (Ipv4RoutesCase) table.getRoutes();
java.util.Map<String, List<IpPrefix>> pids = new LinkedHashMap<>();
for (Ipv4Route route : routesCase.getIpv4Routes().getIpv4Route()) {
List<Segments> segments = route.getAttributes().getAsPath().getSegments();
String pidName = "PID0";
if (segments != null && !segments.isEmpty()) {
List<AsNumber> asSequence = segments.get(segments.size() - 1).getAsSequence();
if (asSequence != null && !asSequence.isEmpty()) {
pidName = "PID" + asSequence.get(asSequence.size() - 1).getValue().toString();
} else {
List<AsNumber> asSet = segments.get(segments.size() - 1).getAsSet();
if (asSet != null && !asSet.isEmpty()) {
pidName = "PID" + String.join("-",
(String[]) Arrays.stream((AsNumber[]) asSet.toArray())
.map(s -> s.getValue().toString())
.toArray());
}
}
}
if (!pids.containsKey(pidName)) {
pids.put(pidName, new LinkedList<>());
}
pids.get(pidName).add(new IpPrefix(route.getPrefix()));
}
for (java.util.Map.Entry<String, List<IpPrefix>> entry : pids.entrySet()) {
String pidName = entry.getKey();
List<IpPrefix> prefixList = entry.getValue();
networkMap.add(new MapBuilder()
.setPid(new PidName(pidName))
.setEndpointAddressGroup(Arrays.asList(new EndpointAddressGroup[]{
new EndpointAddressGroupBuilder()
.setAddressType(new EndpointAddressType(EndpointAddressType.Enumeration.Ipv4))
.setEndpointPrefix(prefixList)
.build()
}))
.build());
}
} else {
throw new ExecutionException("Unsupported route type", null);
}
} else {
throw new InterruptedException("BGP Local RIB not found");
}
} catch (InterruptedException | ExecutionException e) {
throw e;
}
} else {
throw new InterruptedException("Unsupported algorithm for bgp topology");
}
return networkMap;
}

@Override
public void close() throws Exception {
LOG.info("BGP IPv4 updater closed");
}
}
Loading

0 comments on commit f538621

Please sign in to comment.