[acl_loader]: Fix IP match bug for checking IP version (#387)

Add .version to get the version of the IP Signed-off-by: Shu0T1an ChenG <shuche@microsoft.com>
sonic-net · Nov 21, 2018 · a099681 · a099681
1 parent 83ec276
commit a099681
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/acl_loader/main.py b/acl_loader/main.py
@@ -271,14 +271,14 @@ def convert_ip(self, table_name, rule_idx, rule):
 
         if rule.ip.config.source_ip_address:
             source_ip_address = rule.ip.config.source_ip_address.encode("ascii")
-            if ipaddr.IPNetwork(source_ip_address) == 4:
+            if ipaddr.IPNetwork(source_ip_address).version == 4:
                 rule_props["SRC_IP"] = source_ip_address
             else:
                 rule_props["SRC_IPV6"] = source_ip_address
 
         if rule.ip.config.destination_ip_address:
             destination_ip_address = rule.ip.config.destination_ip_address.encode("ascii")
-            if ipaddr.IPNetwork(destination_ip_address) == 4:
+            if ipaddr.IPNetwork(destination_ip_address).version == 4:
                 rule_props["DST_IP"] = destination_ip_address
             else:
                 rule_props["DST_IPV6"] = destination_ip_address