diff options
| -rw-r--r-- | mpm/python/usrp_mpm/sys_utils/net.py | 16 |
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 + } + |