2 * Copyright © 2017 Ericsson India Global Services Pvt Ltd. and others. All rights reserved.
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
8 package org.opendaylight.netvirt.natservice.internal;
10 import java.util.concurrent.ExecutionException;
11 import org.opendaylight.genius.datastoreutils.listeners.DataTreeEventCallbackRegistrar;
12 import org.opendaylight.genius.interfacemanager.interfaces.IInterfaceManager;
13 import org.opendaylight.genius.mdsalutil.interfaces.IMdsalApiManager;
14 import org.opendaylight.mdsal.binding.api.DataBroker;
15 import org.opendaylight.mdsal.binding.util.Datastore.Configuration;
16 import org.opendaylight.mdsal.binding.util.TypedReadWriteTransaction;
17 import org.opendaylight.netvirt.fibmanager.api.IFibManager;
18 import org.opendaylight.netvirt.natservice.ha.NatDataUtil;
19 import org.opendaylight.netvirt.vpnmanager.api.IVpnFootprintService;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.IdManagerService;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.interfacemanager.rpcs.rev160406.OdlInterfaceRpcService;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.itm.rpcs.rev160406.ItmRpcService;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.ProviderTypes;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.natservice.rev160111.ext.routers.Routers;
25 import org.opendaylight.yangtools.yang.common.Uint64;
26 import org.slf4j.Logger;
27 import org.slf4j.LoggerFactory;
29 public class FlatVlanConntrackBasedSnatService extends ConntrackBasedSnatService {
31 private static final Logger LOG = LoggerFactory.getLogger(FlatVlanConntrackBasedSnatService.class);
33 public FlatVlanConntrackBasedSnatService(DataBroker dataBroker, IMdsalApiManager mdsalManager,
34 ItmRpcService itmManager, OdlInterfaceRpcService odlInterfaceRpcService,
35 IdManagerService idManager, NAPTSwitchSelector naptSwitchSelector,
36 IInterfaceManager interfaceManager,
37 IVpnFootprintService vpnFootprintService,
38 IFibManager fibManager, NatDataUtil natDataUtil,
39 DataTreeEventCallbackRegistrar eventCallbacks) {
40 super(dataBroker, mdsalManager, itmManager, idManager, naptSwitchSelector,
41 odlInterfaceRpcService, interfaceManager, vpnFootprintService, fibManager ,
42 natDataUtil, eventCallbacks);
46 public boolean addSnatAllSwitch(TypedReadWriteTransaction<Configuration> confTx, Routers routers,
47 Uint64 primarySwitchId) {
48 if (checkProviderType(confTx, routers)) {
51 return super.addSnatAllSwitch(confTx, routers, primarySwitchId);
55 public boolean removeSnatAllSwitch(TypedReadWriteTransaction<Configuration> confTx, Routers routers,
56 Uint64 primarySwitchId) throws ExecutionException, InterruptedException {
57 return !checkProviderType(confTx, routers) && super.removeSnatAllSwitch(confTx, routers, primarySwitchId);
61 public boolean addSnat(TypedReadWriteTransaction<Configuration> confTx, Routers routers,
62 Uint64 primarySwitchId, Uint64 dpnId) {
63 return !checkProviderType(confTx, routers) && super.addSnat(confTx, routers, primarySwitchId, dpnId);
67 public boolean removeSnat(TypedReadWriteTransaction<Configuration> confTx, Routers routers,
68 Uint64 primarySwitchId, Uint64 dpnId) throws ExecutionException, InterruptedException {
69 return !checkProviderType(confTx, routers) && super.removeSnat(confTx, routers, primarySwitchId, dpnId);
72 private boolean checkProviderType(TypedReadWriteTransaction<Configuration> confTx, Routers routers) {
73 ProviderTypes extNwProviderType = NatUtil.getProviderTypefromNetworkId(confTx, routers.getNetworkId());
74 LOG.debug("FlatVlanConntrackBasedSnatService: ProviderTypes {}", extNwProviderType);
75 return extNwProviderType == ProviderTypes.VXLAN || extNwProviderType == ProviderTypes.GRE;