natservice/impl/src/main/java/org/opendaylight/netvirt/natservice/internal/FlatVlanConntrackBasedSnatService.java

   1 /*
   2  * Copyright © 2017 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.natservice.internal;
   9
  10 import java.math.BigInteger;
  11
  12 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  13 import org.opendaylight.genius.interfacemanager.interfaces.IInterfaceManager;
  14 import org.opendaylight.genius.mdsalutil.interfaces.IMdsalApiManager;
  15 import org.opendaylight.netvirt.fibmanager.api.IFibManager;
  16 import org.opendaylight.netvirt.vpnmanager.api.IVpnFootprintService;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.IdManagerService;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rpcs.rev160406.OdlInterfaceRpcService;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.ItmRpcService;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.ProviderTypes;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.ext.routers.Routers;
  22 import org.slf4j.Logger;
  23 import org.slf4j.LoggerFactory;
  24
  25 public class FlatVlanConntrackBasedSnatService extends ConntrackBasedSnatService {
  26
  27     private static final Logger LOG = LoggerFactory.getLogger(FlatVlanConntrackBasedSnatService.class);
  28
  29     public FlatVlanConntrackBasedSnatService(DataBroker dataBroker, IMdsalApiManager mdsalManager,
  30                                              ItmRpcService itmManager, OdlInterfaceRpcService odlInterfaceRpcService,
  31                                              IdManagerService idManager, NAPTSwitchSelector naptSwitchSelector,
  32                                              IInterfaceManager interfaceManager,
  33                                              IVpnFootprintService vpnFootprintService,
  34                                              IFibManager fibManager) {
  35         super(dataBroker, mdsalManager, itmManager, idManager, naptSwitchSelector,
  36                 odlInterfaceRpcService, interfaceManager, vpnFootprintService, fibManager);
  37     }
  38
  39     @Override
  40     public boolean handleSnatAllSwitch(Routers routers, BigInteger primarySwitchId,  int addOrRemove) {
  41         ProviderTypes extNwProviderType = NatUtil.getProviderTypefromNetworkId(dataBroker, routers.getNetworkId());
  42         LOG.debug("FlatVlanConntrackBasedSnatService: handleSnatAllSwitch ProviderTypes {}", extNwProviderType);
  43         if (extNwProviderType == ProviderTypes.VXLAN || extNwProviderType == ProviderTypes.GRE) {
  44             return false;
  45         }
  46         return super.handleSnatAllSwitch(routers, primarySwitchId, addOrRemove);
  47     }
  48
  49     @Override
  50     public boolean handleSnat(Routers routers, BigInteger primarySwitchId, BigInteger dpnId,  int addOrRemove) {
  51         ProviderTypes extNwProviderType = NatUtil.getProviderTypefromNetworkId(dataBroker, routers.getNetworkId());
  52         LOG.debug("FlatVlanConntrackBasedSnatService: handleSnat ProviderTypes {}", extNwProviderType);
  53         if (extNwProviderType == ProviderTypes.VXLAN || extNwProviderType == ProviderTypes.GRE) {
  54             return false;
  55         }
  56         return super.handleSnat(routers, primarySwitchId, dpnId, addOrRemove);
  57     }
  58 }