aboutsummaryrefslogtreecommitdiffstats
path: root/mpm/python
diff options
context:
space:
mode:
Diffstat (limited to 'mpm/python')
-rw-r--r--mpm/python/usrp_mpm/sys_utils/net.py16
1 files changed, 15 insertions, 1 deletions
diff --git a/mpm/python/usrp_mpm/sys_utils/net.py b/mpm/python/usrp_mpm/sys_utils/net.py
index 2cf4071ab..f40e147dc 100644
--- a/mpm/python/usrp_mpm/sys_utils/net.py
+++ b/mpm/python/usrp_mpm/sys_utils/net.py
@@ -9,7 +9,7 @@ N310 implementation module
import itertools
import socket
from six import iteritems
-from pyroute2 import IPRoute
+from pyroute2 import IPRoute, IPDB
from usrp_mpm.mpmlog import get_logger
@@ -117,3 +117,17 @@ def get_mac_addr(remote_addr):
if not addrs:
return None
return addrs[0].get_attr('NDA_LLADDR')
+
+def get_local_ip_addrs(ipv4_only=False):
+ """
+ Return a set of IP addresses which are bound to local interfaces.
+ """
+ with IPDB() as ipdb:
+ return {
+ ip_subnet[0]
+ for ip_subnet_list
+ in [x['ipaddr'] for x in ipdb.interfaces.values()]
+ for ip_subnet in ip_subnet_list
+ if not ipv4_only or ip_subnet[0].find(':') == -1
+ }
+