class InterfaceManager(object):
def __init__(self, server_handler):
self._device_classes = {}
self._devices = {} #ifindex to device
self._nl_socket = IPRSocket()
self._nl_socket.bind(groups=NL_GROUPS)
self._msg_queue = deque()
#TODO split DevlinkManager away from the InterfaceManager
#self._dl_manager = DevlinkManager()
self._server_handler = server_handler
def clear_dev_classes(self):
self._device_classes = {}
def add_device_class(self, name, cls):
if name in self._device_classes:
raise InterfaceManagerError("Device class name conflict %s" % name)
self._device_classes[name] = cls
return cls
def reconnect_netlink(self):
if self._nl_socket != None:
self._nl_socket.close()
self._nl_socket = None
self._nl_socket = IPRSocket()
self._nl_socket.bind(groups=NL_GROUPS)
self.rescan_devices()
def get_nl_socket(self):
return self._nl_socket
def pull_netlink_messages_into_queue(self):
try:
while True:
rl, wl, xl = select.select([self._nl_socket], [], [], 0)
if not len(rl):
break
self._msg_queue.extend(self._nl_socket.get())
except socket.error:
self.reconnect_netlink()
return []
def rescan_devices(self):
self.request_netlink_dump()
self.handle_netlink_msgs()
def request_netlink_dump(self):
self._nl_socket.put(None,
RTM_GETLINK,
msg_flags=NLM_F_REQUEST | NLM_F_DUMP)
self._nl_socket.put(None,
RTM_GETADDR,
msg_flags=NLM_F_REQUEST | NLM_F_DUMP)
def handle_netlink_msgs(self):
self.pull_netlink_messages_into_queue()
while len(self._msg_queue):
msg = self._msg_queue.popleft()
self._handle_netlink_msg(msg)
# self._dl_manager.rescan_ports()
# for device in self._devices.values():
# dl_port = self._dl_manager.get_port(device.name)
# device._set_devlink(dl_port)
def _handle_netlink_msg(self, msg):
if msg['header']['type'] in [RTM_NEWLINK, RTM_NEWADDR, RTM_DELADDR]:
if msg['index'] in self._devices:
self._devices[msg['index']]._update_netlink(msg)
elif msg['header']['type'] == RTM_NEWLINK:
dev = self._device_classes["Device"](self)
dev._init_netlink(msg)
self._devices[msg['index']] = dev
update_msg = {
"type": "dev_created",
"dev_data": dev._get_if_data()
}
self._server_handler.send_data_to_ctl(update_msg)
dev._disable()
elif msg['header']['type'] == RTM_DELLINK:
if msg['family'] == PF_BRIDGE:
return
if msg['index'] in self._devices:
dev = self._devices[msg['index']]
dev._deleted = True
del self._devices[msg['index']]
del_msg = {"type": "dev_deleted", "ifindex": msg['index']}
self._server_handler.send_data_to_ctl(del_msg)
else:
return
def untrack_device(self, dev):
if dev.ifindex in self._devices:
del self._devices[dev.ifindex]
def get_device(self, ifindex):
self.rescan_devices()
if ifindex in self._devices:
return self._devices[ifindex]
else:
raise DeviceNotFound()
def get_devices(self):
self.rescan_devices()
return list(self._devices.values())
def get_device_by_hwaddr(self, hwaddr):
self.rescan_devices()
for dev in list(self._devices.values()):
if dev.hwaddr == hwaddr:
return dev
raise DeviceNotFound()
def get_device_by_name(self, name):
self.rescan_devices()
for dev in list(self._devices.values()):
if dev.name == name:
return dev
raise DeviceNotFound()
def get_device_by_params(self, params):
self.rescan_devices()
matched = None
for dev in list(self._devices.values()):
matched = dev
dev_data = dev.get_if_data()
for key, value in params.items():
if key not in dev_data or dev_data[key] != value:
matched = None
break
if matched:
break
return matched
def deconfigure_all(self):
for dev in self._devices.values():
pass
# dev.clear_configuration()
def create_device(self, clsname, args=[], kwargs={}):
devcls = self._device_classes[clsname]
try:
device = devcls(self, *args, **kwargs)
except KeyError as e:
raise DeviceConfigError("%s is a mandatory argument" % e)
device._create()
device._bulk_enabled = False
self.request_netlink_dump()
self.pull_netlink_messages_into_queue()
device_found = False
while len(self._msg_queue):
msg = self._msg_queue.popleft()
if msg.get_attr("IFLA_IFNAME") == device.name:
device_found = True
device._init_netlink(msg)
self._devices[msg['index']] = device
else:
self._handle_netlink_msg(msg)
if device_found:
return device
else:
raise DeviceError("Device creation failed")
def replace_dev(self, if_id, dev):
del self._devices[if_id]
self._devices[if_id] = dev
def _is_name_used(self, name):
self.rescan_devices()
for device in self._devices.values():
if name == device.name:
return True
out, _ = exec_cmd("ovs-vsctl --columns=name list Interface",
log_outputs=False,
die_on_err=False)
for line in out.split("\n"):
m = re.match(r'.*: \"(.*)\"', line)
if m is not None:
if name == m.group(1):
return True
return False
def assign_name(self, prefix):
index = 0
while (self._is_name_used(prefix + str(index))):
index += 1
return prefix + str(index)
def _assign_name_pair(self, prefix):
index1 = 0
index2 = 0
while (self._is_name_used(prefix + str(index1))):
index1 += 1
index2 = index1 + 1
while (self._is_name_used(prefix + str(index2))):
index2 += 1
return prefix + str(index1), prefix + str(index2)
