Update DAO API
[lispflowmapping.git] / mappingservice / implementation / src / main / java / org / opendaylight / lispflowmapping / implementation / lisp / MapResolver.java
index 0c40720f80e37a64b77f188677b76f598d913769..298e20c2ca8968428136572a525e3985a1dd3806 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2013 Contextream, Inc. and others.  All rights reserved.
+ * Copyright (c) 2014 Contextream, Inc. 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,
 
 package org.opendaylight.lispflowmapping.implementation.lisp;
 
-import java.util.Map;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Set;
 
+import org.apache.commons.lang3.exception.ExceptionUtils;
+import org.opendaylight.lispflowmapping.implementation.config.ConfigIni;
+import org.opendaylight.lispflowmapping.implementation.mapcache.TopologyMapCache;
 import org.opendaylight.lispflowmapping.interfaces.dao.ILispDAO;
-import org.opendaylight.lispflowmapping.interfaces.dao.IMappingServiceKey;
-import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceKeyUtil;
-import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceRLOC;
-import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceValue;
-import org.opendaylight.lispflowmapping.interfaces.lisp.IMapResolver;
-import org.opendaylight.lispflowmapping.type.lisp.EidRecord;
-import org.opendaylight.lispflowmapping.type.lisp.EidToLocatorRecord;
-import org.opendaylight.lispflowmapping.type.lisp.MapReply;
-import org.opendaylight.lispflowmapping.type.lisp.MapRequest;
-import org.opendaylight.lispflowmapping.type.lisp.address.IMaskable;
+import org.opendaylight.lispflowmapping.interfaces.dao.SubscriberRLOC;
+import org.opendaylight.lispflowmapping.interfaces.lisp.IMapRequestResultHandler;
+import org.opendaylight.lispflowmapping.interfaces.lisp.IMapResolverAsync;
+import org.opendaylight.lispflowmapping.lisp.util.LispAFIConvertor;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.EidToLocatorRecord.Action;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.LcafTrafficEngineeringAddress;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.MapRequest;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidrecords.EidRecord;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidtolocatorrecords.EidToLocatorRecord;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidtolocatorrecords.EidToLocatorRecordBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.lcaftrafficengineeringaddress.Hops;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.lispaddress.LispAddressContainer;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.lispaddress.lispaddresscontainer.address.LcafTrafficEngineering;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.locatorrecords.LocatorRecord;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.locatorrecords.LocatorRecordBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.mapreplymessage.MapReplyBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.maprequest.ItrRloc;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-public class MapResolver implements IMapResolver {
-    private ILispDAO dao;
-    private volatile boolean shouldAuthenticate;
-    private volatile boolean shouldIterateMask;
-    protected static final Logger logger = LoggerFactory.getLogger(MapResolver.class);
+import com.google.common.base.Preconditions;
+import com.google.common.collect.Sets;
+
+public class MapResolver extends AbstractLispComponent implements IMapResolverAsync {
+
+    private static final int TTL_RLOC_TIMED_OUT = 1;
+    private static final int TTL_NO_RLOC_KNOWN = 15;
+    protected static final Logger LOG = LoggerFactory.getLogger(MapResolver.class);
+
+    private static final ConfigIni configIni = new ConfigIni();
+    private static final String elpPolicy = configIni.getElpPolicy();
 
     public MapResolver(ILispDAO dao) {
         this(dao, true, true);
     }
 
     public MapResolver(ILispDAO dao, boolean authenticate, boolean iterateMask) {
-        this.dao = dao;
-        this.shouldAuthenticate = authenticate;
-        this.shouldIterateMask = iterateMask;
+        super(dao, authenticate, iterateMask);
     }
 
-    public MapReply handleMapRequest(MapRequest request) {
-        if (dao == null) {
-            logger.warn("handleMapRequest called while dao is uninitialized");
-            return null;
-        }
-        MapReply mapReply = new MapReply();
-        mapReply.setNonce(request.getNonce());
-
-        for (EidRecord eid : request.getEids()) {
-            EidToLocatorRecord eidToLocators = new EidToLocatorRecord();
-            eidToLocators.setMaskLength(eid.getMaskLength())//
-                    .setPrefix(eid.getPrefix());
-            IMappingServiceKey key = MappingServiceKeyUtil.generateMappingServiceKey(eid.getPrefix(), eid.getMaskLength());
-            Map<String, ?> locators = dao.get(key);
-            if (shouldIterateMask() && locators == null && key.getEID() instanceof IMaskable) {
-                locators = findMaskLocators(key);
-            }
-            if (locators != null) {
-                addLocators(eidToLocators, locators);
-            }
-            mapReply.addEidToLocator(eidToLocators);
-        }
-
-        return mapReply;
-    }
+    public void handleMapRequest(MapRequest request, boolean smr, IMapRequestResultHandler callback) {
+        Preconditions.checkNotNull(dao);
 
-    private Map<String, ?> findMaskLocators(IMappingServiceKey key) {
-        int mask = key.getMask();
-        while (mask > 0) {
-            key = MappingServiceKeyUtil.generateMappingServiceKey(key.getEID(), (byte) mask);
-            ((IMaskable) key.getEID()).normalize(mask);
-            mask--;
-            Map<String, ?> locators = dao.get(key);
-            if (locators != null) {
-                return locators;
+        LispAddressContainer srcEid = null;
+        if (request.getSourceEid() != null) {
+            srcEid = request.getSourceEid().getLispAddressContainer();
+        }
+        MapReplyBuilder builder = new MapReplyBuilder();
+        builder.setEchoNonceEnabled(false);
+        builder.setProbe(false);
+        builder.setSecurityEnabled(false);
+        builder.setNonce(request.getNonce());
+        builder.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
+        for (EidRecord eidRecord : request.getEidRecord()) {
+            EidToLocatorRecord mapping = TopologyMapCache.getMapping(srcEid, eidRecord.getLispAddressContainer(), dao);
+            EidToLocatorRecordBuilder recordBuilder;
+            if (mapping != null) {
+                recordBuilder = new EidToLocatorRecordBuilder(mapping);
+                List<ItrRloc> itrRlocs = request.getItrRloc();
+                if (itrRlocs != null && itrRlocs.size() != 0) {
+                    updateLocators(recordBuilder, itrRlocs);
+                    LispAddressContainer itrRloc = itrRlocs.get(0).getLispAddressContainer();
+                    SubscriberRLOC subscriberRloc = new SubscriberRLOC(itrRloc, srcEid);
+                    Set<SubscriberRLOC> subscribers = TopologyMapCache.getSubscribers(mapping.getLispAddressContainer(), dao);
+                    if (subscribers == null) {
+                        subscribers = Sets.newConcurrentHashSet();
+                    } else if (subscribers.contains(subscriberRloc)) {
+                        // If there is an entry already for this subscriberRloc, remove it, so that it gets the new
+                        // timestamp
+                        subscribers.remove(subscriberRloc);
+                    }
+                    if (smr) {
+                        LOG.trace("Adding new subscriber: " + subscriberRloc.toString());
+                        subscribers.add(subscriberRloc);
+                        TopologyMapCache.addSubscribers(mapping.getLispAddressContainer(), subscribers, dao);
+                    }
+                }
+            } else {
+                recordBuilder = new EidToLocatorRecordBuilder();
+                recordBuilder.setAuthoritative(false);
+                recordBuilder.setMapVersion((short) 0);
+                recordBuilder.setMaskLength(eidRecord.getMask());
+                recordBuilder.setLispAddressContainer(eidRecord.getLispAddressContainer());
+                recordBuilder.setAction(Action.NativelyForward);
+                if (shouldAuthenticate() && TopologyMapCache.getAuthenticationKey(eidRecord.getLispAddressContainer(),
+                        dao, shouldIterateMask()) != null) {
+                    recordBuilder.setRecordTtl(TTL_RLOC_TIMED_OUT);
+                } else {
+                    recordBuilder.setRecordTtl(TTL_NO_RLOC_KNOWN);
+
+                }
             }
+            builder.getEidToLocatorRecord().add(recordBuilder.build());
         }
-        return null;
+        callback.handleMapReply(builder.build());
     }
 
-    private void addLocators(EidToLocatorRecord eidToLocators, Map<String, ?> locators) {
+    // Process locators according to configured policy
+    private void updateLocators(EidToLocatorRecordBuilder recordBuilder, List<ItrRloc> itrRlocs) {
+        List<LocatorRecord> locatorRecords = recordBuilder.getLocatorRecord();
+        recordBuilder.setLocatorRecord(new ArrayList<LocatorRecord>());
         try {
-            MappingServiceValue value = (MappingServiceValue) locators.get("value");
-            for (MappingServiceRLOC rloc : value.getRlocs()) {
-                addLocator(eidToLocators, rloc);
-
+            for (LocatorRecord record : locatorRecords) {
+                LispAddressContainer container = record.getLispAddressContainer();
+
+                // For non-ELP RLOCs, or when ELP policy is default, or itrRlocs is null, just add the locator and be
+                // done
+                if ((!(container.getAddress() instanceof LcafTrafficEngineering))
+                        || elpPolicy.equalsIgnoreCase("default") || itrRlocs == null) {
+                    recordBuilder.getLocatorRecord().add(
+                            new LocatorRecordBuilder().setLocalLocator(record.isLocalLocator())
+                                    .setRlocProbed(record.isRlocProbed()).setWeight(record.getWeight())
+                                    .setPriority(record.getPriority()).setMulticastWeight(record.getMulticastWeight())
+                                    .setMulticastPriority(record.getMulticastPriority()).setRouted(record.isRouted())
+                                    .setLispAddressContainer(container).setName(record.getName()).build());
+                    continue;
+                }
+
+                LcafTrafficEngineeringAddress teAddress = ((LcafTrafficEngineering) container.getAddress())
+                        .getLcafTrafficEngineeringAddr();
+                LispAddressContainer nextHop = getNextELPHop(teAddress, itrRlocs);
+                if (nextHop != null) {
+                    java.lang.Short priority = record.getPriority();
+                    if (elpPolicy.equalsIgnoreCase("both")) {
+                        recordBuilder.getLocatorRecord().add(
+                                new LocatorRecordBuilder().setLocalLocator(record.isLocalLocator())
+                                        .setRlocProbed(record.isRlocProbed()).setWeight(record.getWeight())
+                                        .setPriority(record.getPriority())
+                                        .setMulticastWeight(record.getMulticastWeight())
+                                        .setMulticastPriority(record.getMulticastPriority())
+                                        .setRouted(record.isRouted()).setLispAddressContainer(container)
+                                        .setName(record.getName()).build());
+                        // Make the priority of the added simple locator lower so that ELP is used by default if
+                        // the xTR understands ELP.  Exclude 255, since that means don't use for unicast forwarding
+                        // XXX Complex cases like several ELPs with different priorities are not handled
+                        if (priority != 254 || priority !=255) {
+                            priority++;
+                        }
+                    }
+                    // Build and add the simple RLOC
+                    recordBuilder.getLocatorRecord().add(
+                            new LocatorRecordBuilder().setLocalLocator(record.isLocalLocator())
+                                    .setRlocProbed(record.isRlocProbed()).setWeight(record.getWeight())
+                                    .setPriority(priority).setMulticastWeight(record.getMulticastWeight())
+                                    .setMulticastPriority(record.getMulticastPriority()).setRouted(record.isRouted())
+                                    .setLispAddressContainer(nextHop).setName(record.getName()).build());
+                }
             }
         } catch (ClassCastException cce) {
+            LOG.error("Class Cast Exception while building EidToLocatorRecord: {}", ExceptionUtils.getStackTrace(cce));
         }
     }
 
-    private void addLocator(EidToLocatorRecord eidToLocators, MappingServiceRLOC locatorObject) {
-        if (locatorObject == null) {
-            return;
-        }
-        try {
-            eidToLocators.addLocator(locatorObject.getRecord().clone().setRouted(true));
-            eidToLocators.setAction(locatorObject.getAction());
-            eidToLocators.setAuthoritative(locatorObject.isAuthoritative());
-            eidToLocators.setRecordTtl(locatorObject.getTtl());
-        } catch (ClassCastException cce) {
+    private LispAddressContainer getNextELPHop(LcafTrafficEngineeringAddress elp, List<ItrRloc> itrRlocs) {
+        LispAddressContainer nextHop = null;
+        List<Hops> hops = elp.getHops();
+
+        if (hops != null && hops.size() > 0) {
+            // By default we return the first hop
+            nextHop = LispAFIConvertor.toContainer(LispAFIConvertor.toAFIfromPrimitive(hops.get(0).getHop()
+                    .getPrimitiveAddress()));
+            for (Hops hop : hops) {
+                LispAddressContainer hopContainer = LispAFIConvertor.toContainer(LispAFIConvertor
+                        .toAFIfromPrimitive(hop.getHop().getPrimitiveAddress()));
+                for (ItrRloc itrRloc : itrRlocs) {
+                    if (itrRloc.getLispAddressContainer().equals(hopContainer)) {
+                        int i = hops.indexOf(hop);
+                        if (i < hops.size() - 1) {
+                            nextHop = LispAFIConvertor.toContainer(LispAFIConvertor.toAFIfromPrimitive(hops.get(i + 1)
+                                    .getHop().getPrimitiveAddress()));
+                            return nextHop;
+                        }
+                    }
+                }
+            }
         }
-    }
-
-    public boolean shouldIterateMask() {
-        return shouldIterateMask;
-    }
-
-    public boolean shouldAuthenticate() {
-        return shouldAuthenticate;
-    }
-
-    public void setShouldIterateMask(boolean shouldIterateMask) {
-        this.shouldIterateMask = shouldIterateMask;
-    }
 
-    public void setShouldAuthenticate(boolean shouldAuthenticate) {
-        this.shouldAuthenticate = shouldAuthenticate;
+        return nextHop;
     }
 
 }