neutron-ovsdb/src/main/java/org/opendaylight/groupbasedpolicy/neutron/ovsdb/NeutronGbpFloatingIpListener.java

   1 /*
   2  * Copyright (c) 2015 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.groupbasedpolicy.neutron.ovsdb;
  10
  11 import static com.google.common.base.Preconditions.checkNotNull;
  12
  13 import java.util.ArrayList;
  14 import java.util.List;
  15 import java.util.Map.Entry;
  16
  17 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  18 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  19 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  20 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
  21 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
  22 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  23 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  24 import org.opendaylight.groupbasedpolicy.endpoint.EpKey;
  25 import org.opendaylight.groupbasedpolicy.neutron.ovsdb.util.NeutronOvsdbIidFactory;
  26 import org.opendaylight.groupbasedpolicy.util.DataStoreHelper;
  27 import org.opendaylight.groupbasedpolicy.util.IidFactory;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.L2BridgeDomainId;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoint.fields.L3Address;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.Endpoint;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3Builder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.endpoints.EndpointL3Key;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.neutron.gbp.mapper.rev150513.mappings.floating.ip.association.mappings.internal.ports.by.floating.ip.ports.InternalPortByFloatingIpPort;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.neutron.gbp.mapper.rev150513.mappings.gbp.by.neutron.mappings.endpoints.by.ports.EndpointByPort;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.ofoverlay.rev140528.OfOverlayL3Nat;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.ofoverlay.rev140528.OfOverlayL3NatBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.ofoverlay.rev140528.napt.translations.fields.NaptTranslations;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.ofoverlay.rev140528.napt.translations.fields.NaptTranslationsBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.ofoverlay.rev140528.napt.translations.fields.napt.translations.NaptTranslation;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.ofoverlay.rev140528.napt.translations.fields.napt.translations.NaptTranslationBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.ofoverlay.rev140528.napt.translations.fields.napt.translations.NaptTranslationKey;
  44 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  45 import org.opendaylight.yangtools.yang.binding.DataObject;
  46 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  47 import org.slf4j.Logger;
  48 import org.slf4j.LoggerFactory;
  49
  50 import com.google.common.base.Optional;
  51
  52 public class NeutronGbpFloatingIpListener implements DataChangeListener, AutoCloseable {
  53
  54     private static final Logger LOG = LoggerFactory.getLogger(NeutronGbpFloatingIpListener.class);
  55     private final ListenerRegistration<DataChangeListener> gbpFloatingIpListener;
  56     private final DataBroker dataBroker;
  57
  58     public NeutronGbpFloatingIpListener(DataBroker dataBroker) {
  59         this.dataBroker = checkNotNull(dataBroker);
  60          gbpFloatingIpListener = dataBroker.registerDataChangeListener(LogicalDatastoreType.OPERATIONAL,
  61                 NeutronOvsdbIidFactory.neutronGbpInternalPortByFloatingIpIidWildcard(), this, DataChangeScope.BASE);
  62         LOG.trace("NeutronGbpFloatingIpListener started");
  63     }
  64
  65     @Override
  66     public void onDataChanged(AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
  67
  68         /*
  69          */
  70         for (Entry<InstanceIdentifier<?>, DataObject> entry : change.getCreatedData().entrySet()) {
  71             if (entry.getValue() instanceof InternalPortByFloatingIpPort) {
  72                 InternalPortByFloatingIpPort internalPortByFloatingIp = (InternalPortByFloatingIpPort) entry.getValue();
  73                 processInternalPortByFloatingIp(internalPortByFloatingIp);
  74             }
  75         }
  76
  77         /*
  78          * Updates
  79          */
  80         for (Entry<InstanceIdentifier<?>, DataObject> entry : change.getUpdatedData().entrySet()) {
  81             if (entry.getValue() instanceof InternalPortByFloatingIpPort) {
  82                 InternalPortByFloatingIpPort internalPortByFloatingIp = (InternalPortByFloatingIpPort) entry.getValue();
  83                 processInternalPortByFloatingIp(internalPortByFloatingIp);
  84             }
  85         }
  86
  87         /*
  88          * Deletions
  89          */
  90         for (InstanceIdentifier<?> iid : change.getRemovedPaths()) {
  91             /*
  92              * Remove ?
  93              */
  94         }
  95     }
  96
  97     private void processInternalPortByFloatingIp(InternalPortByFloatingIpPort internalPortByFloatingIp) {
  98         IpAddress natAddress = internalPortByFloatingIp.getFloatingIpPortIpAddress();
  99         IpAddress ipAddress = internalPortByFloatingIp.getInternalPortIpAddress();
 100
 101         ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction();
 102         Optional<EndpointByPort> optEndpointByPort = DataStoreHelper.readFromDs(
 103                 LogicalDatastoreType.OPERATIONAL,
 104                 NeutronOvsdbIidFactory.endpointByPortIid(internalPortByFloatingIp.getInternalPortId()), rTx);
 105         if (optEndpointByPort.isPresent()) {
 106             EpKey l2EpKey = new EpKey(optEndpointByPort.get().getL2Context(), optEndpointByPort.get().getMacAddress());
 107             updateEndpointNat(l2EpKey, ipAddress, natAddress);
 108         } else {
 109             LOG.error("processEpByFloatingIp: Couldn't find EP associated with {}.", internalPortByFloatingIp);
 110         }
 111     }
 112
 113     private void updateEndpointNat(EpKey l2EpKey, IpAddress ipAddress, IpAddress natAddress) {
 114         Endpoint l2Ep;
 115         EndpointL3 l3Ep;
 116         EndpointL3Key l3EpKey = null;
 117         OfOverlayL3NatBuilder l3EpAug = null;
 118
 119         ReadWriteTransaction rwTx = dataBroker.newReadWriteTransaction();
 120
 121         Optional<Endpoint> optL2Ep = DataStoreHelper.readFromDs(LogicalDatastoreType.OPERATIONAL,
 122                 IidFactory.endpointIid((L2BridgeDomainId) l2EpKey.getL2Context(), l2EpKey.getMacAddress()), rwTx);
 123         if (optL2Ep.isPresent()) {
 124             l2Ep = optL2Ep.get();
 125         } else {
 126             LOG.error("updateEndpointNat: No Endpoint {} ", l2EpKey);
 127             return;
 128         }
 129
 130         if (l2Ep.getL3Address() == null) {
 131             LOG.error("updateEndpointNat: L2Ep {} had no IP address to translate to.", l2Ep);
 132             return;
 133         }
 134
 135         for (L3Address l3Address : l2Ep.getL3Address()) {
 136             if (l3Address.getIpAddress().equals(ipAddress)) {
 137                 l3EpKey = new EndpointL3Key(l3Address.getIpAddress(), l3Address.getL3Context());
 138                 break;
 139             }
 140         }
 141         if (l3EpKey == null) {
 142             return;
 143         }
 144         Optional<EndpointL3> optL3Ep = DataStoreHelper.readFromDs(LogicalDatastoreType.OPERATIONAL,
 145                 IidFactory.l3EndpointIid(l3EpKey.getL3Context(), l3EpKey.getIpAddress()), rwTx);
 146         if (optL3Ep.isPresent()) {
 147             l3Ep = optL3Ep.get();
 148             OfOverlayL3Nat ofL3NatAug = l3Ep.getAugmentation(OfOverlayL3Nat.class);
 149             if (ofL3NatAug != null) {
 150                 l3EpAug = new OfOverlayL3NatBuilder(ofL3NatAug);
 151             } else {
 152                 l3EpAug = new OfOverlayL3NatBuilder();
 153             }
 154             NaptTranslation nat = new NaptTranslationBuilder().setIpAddress(ipAddress)
 155                 .setKey(new NaptTranslationKey(natAddress))
 156                 .build();
 157             List<NaptTranslation> natList = new ArrayList<>();
 158             natList.add(nat);
 159             NaptTranslations nats = new NaptTranslationsBuilder().setNaptTranslation(natList).build();
 160
 161             EndpointL3 updatedEpL3 = new EndpointL3Builder(l3Ep).addAugmentation(OfOverlayL3Nat.class,
 162                     l3EpAug.setNaptTranslations(nats).build()).build();
 163
 164             rwTx.put(LogicalDatastoreType.OPERATIONAL, IidFactory.l3EndpointIid(l3EpKey.getL3Context(), l3EpKey.getIpAddress()), updatedEpL3,true);
 165             boolean writeResult = DataStoreHelper.submitToDs(rwTx);
 166             if(!writeResult) {
 167                 LOG.trace("updateEndpointNat: Could not write {} to datastore.",updatedEpL3.getKey());
 168             }
 169         }
 170     }
 171
 172     @Override
 173     public void close() throws Exception {
 174         gbpFloatingIpListener.close();
 175     }
 176
 177 }