mappingservice/implementation/src/main/java/org/opendaylight/lispflowmapping/implementation/MappingSystem.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc.  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.lispflowmapping.implementation;
  10
  11 import java.util.ArrayList;
  12 import java.util.EnumMap;
  13 import java.util.List;
  14
  15 import org.opendaylight.lispflowmapping.implementation.config.ConfigIni;
  16 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
  17 import org.opendaylight.lispflowmapping.implementation.mapcache.FlatMapCache;
  18 import org.opendaylight.lispflowmapping.implementation.mapcache.MultiTableMapCache;
  19 import org.opendaylight.lispflowmapping.implementation.mapcache.SimpleMapCache;
  20 import org.opendaylight.lispflowmapping.implementation.mdsal.DataStoreBackEnd;
  21 import org.opendaylight.lispflowmapping.implementation.util.MappingMergeUtil;
  22 import org.opendaylight.lispflowmapping.interfaces.dao.ILispDAO;
  23 import org.opendaylight.lispflowmapping.interfaces.mapcache.IMapCache;
  24 import org.opendaylight.lispflowmapping.interfaces.mapcache.IMappingSystem;
  25 import org.opendaylight.lispflowmapping.lisp.util.LispAddressStringifier;
  26 import org.opendaylight.lispflowmapping.lisp.util.LispAddressUtil;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.SimpleAddress;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.ExplicitLocatorPath;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv4;
  30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv6;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.ServicePath;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.explicit.locator.path.explicit.locator.path.Hop;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.locatorrecords.LocatorRecord;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.locatorrecords.LocatorRecordBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecordBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.rloc.container.Rloc;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.AuthenticationKey;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.Mapping;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.mapping.authkey.container.MappingAuthkey;
  43 import org.slf4j.Logger;
  44 import org.slf4j.LoggerFactory;
  45
  46 /**
  47  * The Mapping System coordinates caching of md-sal stored mappings and if so configured enables longest prefix match
  48  * mapping lookups
  49  *
  50  * @author Florin Coras
  51  *
  52  */
  53 public class MappingSystem implements IMappingSystem {
  54     private final static Logger LOG = LoggerFactory.getLogger(MappingSystem.class);
  55     private boolean iterateMask;
  56     private boolean notificationService;
  57     private boolean overwrite;
  58     private ILispDAO dao;
  59     private IMapCache smc;
  60     private IMapCache pmc;
  61     private final EnumMap<MappingOrigin, IMapCache> tableMap = new EnumMap<>(MappingOrigin.class);
  62     private DataStoreBackEnd dsbe;
  63     private BindingAwareBroker bindingAwareBroker;
  64
  65     public MappingSystem(ILispDAO dao, boolean iterateMask, boolean notifications, boolean overwrite) {
  66         this.dao = dao;
  67         this.iterateMask = iterateMask;
  68         this.notificationService = notifications;
  69         this.overwrite = overwrite;
  70         buildMapCaches();
  71     }
  72
  73     public void setDataStoreBackEnd(DataStoreBackEnd dsbe) {
  74         this.dsbe = dsbe;
  75     }
  76
  77     @Override
  78     public void setOverwritePolicy(boolean overwrite) {
  79         this.overwrite = overwrite;
  80     }
  81
  82     @Override
  83     public void setIterateMask(boolean iterate) {
  84         this.iterateMask = iterate;
  85         if (smc != null || pmc != null) {
  86             buildMapCaches();
  87         }
  88     }
  89
  90     public void initialize() {
  91         restoreDaoFromDatastore();
  92     }
  93
  94     private void buildMapCaches() {
  95         /*
  96          * There exists a direct relationship between MappingOrigins and the tables that are part of the MappingSystem.
  97          * Therefore, if a new origin is added, probably a new table should be instantiate here as well.
  98          */
  99         if (iterateMask) {
 100             smc = new SimpleMapCache(dao.putTable(MappingOrigin.Southbound.toString()));
 101             pmc = new MultiTableMapCache(dao.putTable(MappingOrigin.Northbound.toString()));
 102         } else {
 103             smc = new FlatMapCache(dao.putTable(MappingOrigin.Southbound.toString()));
 104             pmc = new FlatMapCache(dao.putTable(MappingOrigin.Northbound.toString()));
 105         }
 106         tableMap.put(MappingOrigin.Northbound, pmc);
 107         tableMap.put(MappingOrigin.Southbound, smc);
 108     }
 109
 110     public void addMapping(MappingOrigin origin, Eid key, Object value) {
 111         tableMap.get(origin).addMapping(key, value, origin == MappingOrigin.Southbound ? overwrite : true);
 112     }
 113
 114     public void updateMappingRegistration(MappingOrigin origin, Eid key) {
 115         tableMap.get(origin).updateMappingRegistration(key);
 116     }
 117
 118     private MappingRecord updateServicePathMappingRecord(MappingRecord mapping, Eid eid) {
 119         // keep properties of original record
 120         MappingRecordBuilder recordBuilder = new MappingRecordBuilder(mapping);
 121         recordBuilder.setLocatorRecord(new ArrayList<LocatorRecord>());
 122
 123         // there should only be one locator record
 124         if (mapping.getLocatorRecord().size() != 1) {
 125             LOG.warn("MappingRecord associated to ServicePath EID has more than one locator!");
 126             return mapping;
 127         }
 128
 129         LocatorRecord locatorRecord = mapping.getLocatorRecord().get(0);
 130         long serviceIndex = ((ServicePath) eid.getAddress()).getServicePath().getServiceIndex();
 131         int index = LispAddressUtil.STARTING_SERVICE_INDEX - (int) serviceIndex;
 132         Rloc rloc = locatorRecord.getRloc();
 133         if (rloc.getAddress() instanceof Ipv4 || rloc.getAddress() instanceof Ipv6) {
 134             if (index != 0) {
 135                 LOG.warn("Service Index should be 255 for simple IP RLOCs!");
 136             }
 137             return mapping;
 138         } else if (rloc.getAddress() instanceof ExplicitLocatorPath) {
 139             ExplicitLocatorPath elp = (ExplicitLocatorPath) rloc.getAddress();
 140             List<Hop> hops = elp.getExplicitLocatorPath().getHop();
 141
 142             if (index < 0 || index > hops.size())  {
 143                 LOG.warn("Service Index out of bounds!");
 144                 return mapping;
 145             }
 146
 147             SimpleAddress nextHop = hops.get(index).getAddress();
 148             LocatorRecordBuilder lrb = new LocatorRecordBuilder(locatorRecord);
 149             lrb.setRloc(LispAddressUtil.toRloc(nextHop));
 150             recordBuilder.getLocatorRecord().add(lrb.build());
 151             return recordBuilder.build();
 152         } else {
 153             LOG.warn("Nothing to do with ServicePath mapping record");
 154             return mapping;
 155         }
 156     }
 157
 158     @Override
 159     public Object getMapping(Eid src, Eid dst) {
 160         // NOTE: Currently we have two lookup algorithms implemented, which are configurable
 161         // using lookupPolicy in ConfigIni.java
 162
 163         if (ConfigIni.getInstance().getLookupPolicy() == ConfigIni.NB_AND_SB) {
 164             return getMappingNbSbIntersection(src, dst);
 165         } else {
 166             return getMappingNbFirst(src, dst);
 167         }
 168     }
 169
 170     private Object getMappingNbFirst(Eid src, Eid dst) {
 171
 172         // Default lookup policy is northboundFirst
 173         //lookupPolicy == NB_FIRST
 174
 175         Object nbMapping = pmc.getMapping(src, dst);
 176
 177         if (nbMapping == null) {
 178             return smc.getMapping(src, dst);
 179         }
 180         if (dst.getAddress() instanceof ServicePath) {
 181             return updateServicePathMappingRecord((MappingRecord) nbMapping, dst);
 182         }
 183         return nbMapping;
 184     }
 185
 186     private Object getMappingNbSbIntersection(Eid src, Eid dst) {
 187         //lookupPolicy == NB_AND_SB, we return intersection
 188         //of NB and SB mappings, or NB mapping if intersection is empty.
 189
 190         Object nbMapping = pmc.getMapping(src, dst);
 191         if (nbMapping == null) {
 192             return nbMapping;
 193         }
 194         // no intersection for Service Path mappings
 195         if (dst.getAddress() instanceof ServicePath) {
 196             return updateServicePathMappingRecord((MappingRecord)nbMapping, dst);
 197         }
 198         Object sbMapping = smc.getMapping(src, dst);
 199         if (sbMapping == null) {
 200             return nbMapping;
 201         }
 202         // both NB and SB mappings exist. Compute intersection of the mappings
 203         return MappingMergeUtil.computeNbSbIntersection((MappingRecord)nbMapping, (MappingRecord)sbMapping);
 204     }
 205
 206     @Override
 207     public Object getMapping(Eid dst) {
 208         return getMapping((Eid)null, dst);
 209     }
 210
 211     @Override
 212     public Object getMapping(MappingOrigin origin, Eid key) {
 213         return tableMap.get(origin).getMapping(null, key);
 214     }
 215
 216     @Override
 217     public void removeMapping(MappingOrigin origin, Eid key) {
 218         tableMap.get(origin).removeMapping(key, origin == MappingOrigin.Southbound ? overwrite : true);
 219         if (notificationService) {
 220             // TODO
 221         }
 222     }
 223
 224     @Override
 225     public void addAuthenticationKey(Eid key, MappingAuthkey authKey) {
 226         LOG.debug("Adding authentication key '{}' for {}", authKey.getKeyString(),
 227                 LispAddressStringifier.getString(key));
 228         smc.addAuthenticationKey(key, authKey);
 229     }
 230
 231     @Override
 232     public MappingAuthkey getAuthenticationKey(Eid key) {
 233         LOG.debug("Retrieving authentication key for {}", LispAddressStringifier.getString(key));
 234         return smc.getAuthenticationKey(key);
 235     }
 236
 237     @Override
 238     public void removeAuthenticationKey(Eid key) {
 239         LOG.debug("Removing authentication key for {}", LispAddressStringifier.getString(key));
 240         smc.removeAuthenticationKey(key);
 241     }
 242
 243     @Override
 244     public void addData(MappingOrigin origin, Eid key, String subKey, Object data) {
 245         LOG.debug("Add data of class {} for key {} and subkey {}", data.getClass(),
 246                 LispAddressStringifier.getString(key), subKey);
 247         tableMap.get(origin).addData(key, subKey, data);
 248     }
 249
 250     @Override
 251     public Object getData(MappingOrigin origin, Eid key, String subKey) {
 252         LOG.debug("Retrieving data for key {} and subkey {}", LispAddressStringifier.getString(key), subKey);
 253         return tableMap.get(origin).getData(key, subKey);
 254     }
 255
 256     @Override
 257     public void removeData(MappingOrigin origin, Eid key, String subKey) {
 258         LOG.debug("Removing data for key {} and subkey {}", LispAddressStringifier.getString(key), subKey);
 259         tableMap.get(origin).removeData(key, subKey);
 260     }
 261
 262
 263     /**
 264      * Restore all mappings and keys from mdsal datastore
 265      */
 266     private void restoreDaoFromDatastore() {
 267         List<Mapping> mappings = dsbe.getAllMappings();
 268         List<AuthenticationKey> authKeys = dsbe.getAllAuthenticationKeys();
 269
 270         LOG.info("Restoring {} mappings and {} keys from datastore into DAO", mappings.size(), authKeys.size());
 271
 272         for (Mapping mapping : mappings) {
 273             addMapping(mapping.getOrigin(), mapping.getMappingRecord().getEid(), mapping.getMappingRecord());
 274         }
 275
 276         for (AuthenticationKey authKey : authKeys) {
 277             addAuthenticationKey(authKey.getEid(), authKey.getMappingAuthkey());
 278         }
 279     }
 280
 281     public void destroy() {
 282         LOG.info("Mapping System is being destroyed!");
 283     }
 284
 285     @Override
 286     public String printMappings() {
 287         final StringBuffer sb = new StringBuffer();
 288         sb.append("PolicyMapCache\n--------------\n");
 289         sb.append(pmc.printMappings());
 290         sb.append("SbMapCache\n----------\n");
 291         sb.append(smc.printMappings());
 292         return sb.toString();
 293     }
 294
 295     public void cleanCaches() {
 296         dao.removeAll();
 297         buildMapCaches();
 298     }
 299 }