vpnservice/vpnmanager/vpnmanager-impl/src/main/java/org/opendaylight/netvirt/vpnmanager/ArpScheduler.java

   1 /*
   2  * Copyright (c) 2015 Ericsson India Global Services Pvt Ltd. 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.netvirt.vpnmanager;
   9
  10 import com.google.common.base.Optional;
  11 import com.google.common.util.concurrent.CheckedFuture;
  12 import com.google.common.util.concurrent.FutureCallback;
  13 import com.google.common.util.concurrent.ThreadFactoryBuilder;
  14 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  15 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
  16 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  17 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  18 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  19 import org.opendaylight.genius.datastoreutils.AsyncDataTreeChangeListenerBase;
  20 import org.opendaylight.genius.datastoreutils.DataStoreJobCoordinator;
  21 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.VpnAfConfig;
  22 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.VpnInstances;
  23 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.instances.VpnInstance;
  24 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.instances.VpnInstanceKey;
  25 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.interfaces.VpnInterface;
  26 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.interfaces.VpnInterfaceBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.interfaces.VpnInterfaceKey;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rpcs.rev160406.OdlInterfaceRpcService;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.FibEntries;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.fibentries.VrfTables;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.fibentries.VrfTablesKey;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.vrfentries.VrfEntry;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.fibmanager.rev150330.vrfentries.VrfEntryKey;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.Adjacencies;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.adjacency.list.Adjacency;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.l3vpn.rev130911.adjacency.list.AdjacencyKey;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.rev150602.NeutronVpnPortipPortData;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.rev150602.neutron.vpn.portip.port.data.VpnPortipToPort;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.rev150602.neutron.vpn.portip.port.data.VpnPortipToPortKey;
  41 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  43 import org.slf4j.Logger;
  44 import org.slf4j.LoggerFactory;
  45
  46 import java.lang.Thread.UncaughtExceptionHandler;
  47 import java.net.InetAddress;
  48 import java.util.ArrayList;
  49 import java.util.Collection;
  50 import java.util.List;
  51 import java.util.concurrent.*;
  52
  53 public class ArpScheduler extends AsyncDataTreeChangeListenerBase<VpnPortipToPort,ArpScheduler> {
  54
  55     private ScheduledExecutorService executorService;
  56     private OdlInterfaceRpcService interfaceRpc;
  57     private DataBroker dataBroker;
  58     private ScheduledFuture<?> scheduledResult;
  59     private static final Logger LOG = LoggerFactory.getLogger(ArpScheduler.class);
  60     private DelayQueue<MacEntry> macEntryQueue = new DelayQueue<MacEntry>();
  61     private ListenerRegistration<ArpScheduler> listenerRegistration;
  62
  63     public ArpScheduler(OdlInterfaceRpcService interfaceRpc, DataBroker dataBroker) {
  64         super(VpnPortipToPort.class, ArpScheduler.class);
  65         this.dataBroker = dataBroker;
  66         this.interfaceRpc = interfaceRpc;
  67         Long timeout = Long.getLong("arp.cache.timeout");
  68         if (timeout == null) {
  69             timeout = 60000L;
  70         }
  71         ArpConstants.arpCacheTimeout = timeout;
  72         registerListener(this.dataBroker);
  73         executorService = Executors.newScheduledThreadPool(ArpConstants.THREAD_POOL_SIZE, getThreadFactory("Arp Cache Timer Tasks"));
  74         scheduleExpiredEntryDrainerTask();
  75     }
  76
  77
  78     private void scheduleExpiredEntryDrainerTask() {
  79         LOG.info("Scheduling expired entry drainer task");
  80         ExpiredEntryDrainerTask expiredEntryDrainerTask = new ExpiredEntryDrainerTask();
  81         scheduledResult = executorService.scheduleAtFixedRate(expiredEntryDrainerTask, ArpConstants.NO_DELAY, ArpConstants.PERIOD, TimeUnit.MILLISECONDS);
  82     }
  83
  84
  85     private ThreadFactory getThreadFactory(String threadNameFormat) {
  86         ThreadFactoryBuilder builder = new ThreadFactoryBuilder();
  87         builder.setNameFormat(threadNameFormat);
  88         builder.setUncaughtExceptionHandler( new UncaughtExceptionHandler() {
  89             @Override
  90             public void uncaughtException(Thread t, Throwable e) {
  91                 LOG.error("Received Uncaught Exception event in Thread: {}", t.getName(), e);
  92           }
  93        });
  94       return builder.build();
  95      }
  96
  97
  98     private class ExpiredEntryDrainerTask implements Runnable {
  99        @Override
 100        public void run() {
 101            Collection<MacEntry> expiredMacEntries = new ArrayList<>();
 102            macEntryQueue.drainTo(expiredMacEntries);
 103            for (MacEntry macEntry: expiredMacEntries) {
 104                LOG.info("Removing the ARP cache for"+macEntry);
 105                InstanceIdentifier<VpnPortipToPort> id = getVpnPortipToPortInstanceOpDataIdentifier(macEntry.getIpAddress().getHostAddress(),macEntry.getVpnName());
 106                Optional<VpnPortipToPort> vpnPortipToPort = VpnUtil.read(dataBroker, LogicalDatastoreType.CONFIGURATION, id);
 107                if (vpnPortipToPort.isPresent()) {
 108                    VpnPortipToPort vpnPortipToPortold = vpnPortipToPort.get();
 109                    String fixedip = vpnPortipToPortold.getPortFixedip();
 110                    String vpnName =  vpnPortipToPortold.getVpnName();
 111                    String interfaceName =  vpnPortipToPortold.getPortName();
 112                    String rd = getRouteDistinguisher(vpnName);
 113                    DataStoreJobCoordinator coordinator = DataStoreJobCoordinator.getInstance();
 114                    coordinator.enqueueJob(buildJobKey(fixedip,vpnName), new ArpremovechacheTask(dataBroker,fixedip, vpnName,interfaceName, rd,id));
 115                }
 116
 117            }
 118        }
 119     }
 120
 121     private String getRouteDistinguisher(String vpnName) {
 122         InstanceIdentifier<VpnInstance> id = InstanceIdentifier.builder(VpnInstances.class)
 123                  .child(VpnInstance.class, new VpnInstanceKey(vpnName)).build();
 124         Optional<VpnInstance> vpnInstance = VpnUtil.read(dataBroker, LogicalDatastoreType.CONFIGURATION, id);
 125         String rd = "";
 126         if(vpnInstance.isPresent()) {
 127             VpnInstance instance = vpnInstance.get();
 128             VpnAfConfig config = instance.getIpv4Family();
 129             rd = config.getRouteDistinguisher();
 130          }
 131          return rd;
 132     }
 133
 134      public static InstanceIdentifier<VpnPortipToPort> getVpnPortipToPortInstanceOpDataIdentifier(String ip, String vpnName) {
 135            return InstanceIdentifier.builder(NeutronVpnPortipPortData.class)
 136                .child(VpnPortipToPort.class, new VpnPortipToPortKey(ip,vpnName)).build();
 137     }
 138
 139      private void registerListener(final DataBroker db) {
 140        try {
 141             final DataTreeIdentifier<VpnPortipToPort> treeId = new DataTreeIdentifier<>(LogicalDatastoreType.CONFIGURATION, getWildCardPath());
 142             listenerRegistration = dataBroker.registerDataTreeChangeListener(treeId, this);
 143             } catch (final Exception e) {
 144              LOG.error("VPN Service DataChange listener registration fail !", e);
 145              throw new IllegalStateException("VPN Service registration Listener failed.", e);
 146         }
 147        }
 148
 149      @Override
 150        protected ArpScheduler getDataTreeChangeListener() {
 151         return this;
 152     }
 153
 154     @Override
 155     protected void update(InstanceIdentifier<VpnPortipToPort> id, VpnPortipToPort value,
 156             VpnPortipToPort dataObjectModificationAfter) {
 157         try {
 158             InetAddress srcInetAddr = InetAddress.getByName(value.getPortFixedip());
 159             MacAddress srcMacAddress = MacAddress.getDefaultInstance(value.getMacAddress());
 160             String vpnName =  value.getVpnName();
 161             String interfaceName =  value.getPortName();
 162             Boolean islearnt = value.isLearnt();
 163             if(islearnt)
 164             {
 165                 DataStoreJobCoordinator coordinator = DataStoreJobCoordinator.getInstance();
 166                 coordinator.enqueueJob(buildJobKey(srcInetAddr.toString(),vpnName), new ArpaddchacheTask(srcInetAddr, srcMacAddress, vpnName,interfaceName, macEntryQueue));
 167             }
 168         } catch (Exception e) {
 169             LOG.error("Error in deserializing packet {} with exception {}", value, e);
 170             e.printStackTrace();
 171
 172         }
 173     }
 174
 175
 176     @Override
 177     protected void add(InstanceIdentifier<VpnPortipToPort> identifier, VpnPortipToPort value) {
 178
 179         try {
 180             InetAddress srcInetAddr = InetAddress.getByName(value.getPortFixedip());
 181             MacAddress srcMacAddress = MacAddress.getDefaultInstance(value.getMacAddress());
 182             String vpnName =  value.getVpnName();
 183             String interfaceName =  value.getPortName();
 184             Boolean islearnt = value.isLearnt();
 185             if(islearnt)
 186             {
 187                 DataStoreJobCoordinator coordinator = DataStoreJobCoordinator.getInstance();
 188                 if(islearnt)
 189                 {
 190                     coordinator.enqueueJob(buildJobKey(srcInetAddr.toString(),vpnName), new ArpaddchacheTask(srcInetAddr, srcMacAddress, vpnName,interfaceName, macEntryQueue));
 191                 }
 192             }
 193         }
 194         catch (Exception e) {
 195              LOG.error("Error in deserializing packet {} with exception {}", value, e);
 196         }
 197
 198     }
 199
 200     @Override
 201     protected InstanceIdentifier<VpnPortipToPort> getWildCardPath() {
 202        return InstanceIdentifier.create(NeutronVpnPortipPortData.class).child(VpnPortipToPort.class);
 203     }
 204
 205     @Override
 206     protected void remove(InstanceIdentifier<VpnPortipToPort> key, VpnPortipToPort dataObjectModification) {
 207        // TODO Auto-generated method stub
 208
 209      }
 210
 211     private String buildJobKey(String ip, String vpnName){
 212         return new StringBuilder(ArpConstants.ARPJOB).append(ip).append(vpnName).toString();
 213
 214     }
 215 }