aboutsummaryrefslogtreecommitdiffstats
path: root/resources/libraries/python/IPUtil.py
diff options
context:
space:
mode:
Diffstat (limited to 'resources/libraries/python/IPUtil.py')
-rw-r--r--resources/libraries/python/IPUtil.py19
1 files changed, 19 insertions, 0 deletions
diff --git a/resources/libraries/python/IPUtil.py b/resources/libraries/python/IPUtil.py
index 8ad01c491b..00a7000c65 100644
--- a/resources/libraries/python/IPUtil.py
+++ b/resources/libraries/python/IPUtil.py
@@ -12,6 +12,7 @@
# limitations under the License.
"""Common IP utilities library."""
+from ipaddress import IPv4Network
from resources.libraries.python.ssh import SSH
from resources.libraries.python.constants import Constants
@@ -40,3 +41,21 @@ class IPUtil(object):
if int(ret_code) != 0:
raise Exception('VPP ip probe {dev} {ip} failed on {h}'.format(
dev=interface, ip=addr, h=node['host']))
+
+
+def convert_ipv4_netmask_prefix(netmask):
+ """Convert network mask to equivalent network prefix length or vice versa.
+
+ Example: mask 255.255.0.0 -> prefix length 16
+ :param netmask: network mask or network prefix length.
+ :type netmask: str or int
+ :return: network mask or network prefix length.
+ :rtype: str or int
+ """
+ temp_address = "0.0.0.0"
+ net = IPv4Network(u"{0}/{1}".format(temp_address, netmask), False)
+
+ if isinstance(netmask, int):
+ return net.netmask
+ elif isinstance(netmask, basestring):
+ return net.prefixlen