From 82317efb75cd02ec97de7958891b56855fb5a27a Mon Sep 17 00:00:00 2001
From: asarfaty <asarfaty@vmware.com>
Date: Thu, 4 Jun 2020 11:15:19 +0200
Subject: [PATCH] Add feature for supporting spoofguard ipv4 CIDR

Change-Id: Ib0d1d713fa554b462d332a3168675d10f12d9c0f
---
 vmware_nsxlib/v3/__init__.py        | 4 +++-
 vmware_nsxlib/v3/nsx_constants.py   | 1 +
 vmware_nsxlib/v3/policy/__init__.py | 6 ++++++
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/vmware_nsxlib/v3/__init__.py b/vmware_nsxlib/v3/__init__.py
index d58cc101..847d5bc4 100644
--- a/vmware_nsxlib/v3/__init__.py
+++ b/vmware_nsxlib/v3/__init__.py
@@ -170,7 +170,9 @@ class NsxLib(lib.NsxLibBase):
         if (version.LooseVersion(self.get_version()) >=
             version.LooseVersion(nsx_constants.NSX_VERSION_3_1_0)):
             # features available since 3.1.0
-            if (feature == nsx_constants.FEATURE_MP2P_MIGRATION):
+            if feature == nsx_constants.FEATURE_MP2P_MIGRATION:
+                return True
+            if feature == nsx_constants.FEATURE_SPOOFGUARD_CIDR:
                 return True
 
         if (version.LooseVersion(self.get_version()) >=
diff --git a/vmware_nsxlib/v3/nsx_constants.py b/vmware_nsxlib/v3/nsx_constants.py
index f3899e67..be7c300a 100644
--- a/vmware_nsxlib/v3/nsx_constants.py
+++ b/vmware_nsxlib/v3/nsx_constants.py
@@ -176,6 +176,7 @@ FEATURE_GET_TZ_FROM_SWITCH = 'Get TZ endpoints from host switch'
 FEATURE_CONTAINER_CLUSTER_INVENTORY = 'Container Cluster Inventory'
 FEATURE_IPV6 = 'IPV6 Forwarding and Address Allocation'
 FEATURE_MP2P_MIGRATION = 'MP to Policy Migration'
+FEATURE_SPOOFGUARD_CIDR = 'Spoofguard IPv4 CIDR'
 
 # Features available depending on the Policy Manager backend version
 FEATURE_NSX_POLICY = 'NSX Policy'
diff --git a/vmware_nsxlib/v3/policy/__init__.py b/vmware_nsxlib/v3/policy/__init__.py
index b5f87301..76b68e68 100644
--- a/vmware_nsxlib/v3/policy/__init__.py
+++ b/vmware_nsxlib/v3/policy/__init__.py
@@ -185,6 +185,12 @@ class NsxPolicyLib(lib.NsxLibBase):
             if (feature == nsx_constants.FEATURE_RELAX_SCALE_VALIDATION):
                 return True
 
+        if (version.LooseVersion(self.get_version()) >=
+            version.LooseVersion(nsx_constants.NSX_VERSION_3_1_0)):
+            # features available since 3.1.0
+            if feature == nsx_constants.FEATURE_SPOOFGUARD_CIDR:
+                return True
+
         return (feature == nsx_constants.FEATURE_NSX_POLICY)
 
     def reinitialize_cluster(self, resource, event, trigger, payload=None):