# Copyright 2015 Canonical, Ltd. # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as # published by the Free Software Foundation, either version 3 of the # License, or (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . """ Network Model Provides network device listings and extended network information """ import logging from socket import AF_INET, AF_INET6 from urwid import ( connect_signal, LineBox, ProgressBar, Text, ) from subiquitycore.models.network import NetDevAction from subiquitycore.ui.actionmenu import ActionMenu from subiquitycore.ui.buttons import ( back_btn, cancel_btn, done_btn, ) from subiquitycore.ui.container import ( ListBox, Pile, WidgetWrap, ) from subiquitycore.ui.stretchy import StretchyOverlay from subiquitycore.ui.table import ColSpec, TablePile, TableRow from subiquitycore.ui.utils import ( button_pile, Color, make_action_menu_row, Padding, ) from .network_configure_manual_interface import ( AddBondStretchy, AddVlanStretchy, EditNetworkStretchy, ViewInterfaceInfo, ) from .network_configure_wlan_interface import NetworkConfigureWLANStretchy from subiquitycore.view import BaseView log = logging.getLogger('subiquitycore.views.network') class ApplyingConfigWidget(WidgetWrap): def __init__(self, step_count, cancel_func): self.cancel_func = cancel_func button = cancel_btn(_("Cancel"), on_press=self.do_cancel) self.bar = ProgressBar(normal='progress_incomplete', complete='progress_complete', current=0, done=step_count) box = LineBox(Pile([self.bar, button_pile([button])]), title=_("Applying network config")) super().__init__(box) def advance(self): self.bar.current += 1 def do_cancel(self, sender): self.cancel_func() def _stretchy_shower(cls, *args): def impl(self, device): self.show_stretchy_overlay(cls(self, device, *args)) impl.opens_dialog = True return impl class NetworkView(BaseView): title = _("Network connections") excerpt = _("Configure at least one interface this server can use to talk " "to other machines, and which preferably provides sufficient " "access for updates.") footer = _("Select an interface to configure it or select Done to " "continue") def __init__(self, model, controller): self.model = model self.controller = controller self.dev_to_row = {} self.cur_netdevs = [] self.error = Text("", align='center') self.device_table = TablePile( self._build_model_inputs(), spacing=2, colspecs={ 0: ColSpec(rpad=1), 4: ColSpec(can_shrink=True, rpad=1), }) self.listbox = ListBox([self.device_table] + [ Padding.line_break(""), ]) self.bottom = Pile([ Text(""), self._build_buttons(), Text(""), ]) self.error_showing = False self.frame = Pile([ ('pack', Text("")), ('pack', Padding.center_79(Text(_(self.excerpt)))), ('pack', Text("")), Padding.center_90(self.listbox), ('pack', self.bottom)]) self.frame.focus_position = 4 super().__init__(self.frame) def _build_buttons(self): back = back_btn(_("Back"), on_press=self.cancel) done = done_btn(_("Done"), on_press=self.done) return button_pile([done, back]) _action_INFO = _stretchy_shower(ViewInterfaceInfo) _action_EDIT_WLAN = _stretchy_shower(NetworkConfigureWLANStretchy) _action_EDIT_IPV4 = _stretchy_shower(EditNetworkStretchy, 4) _action_EDIT_IPV6 = _stretchy_shower(EditNetworkStretchy, 6) _action_ADD_VLAN = _stretchy_shower(AddVlanStretchy) _action_ADD_BOND = _stretchy_shower(AddBondStretchy) def _action_DELETE(self, device): self.controller.rm_virtual_interface(device) def _action(self, sender, action, device): action, meth = action log.debug("_action %s %s", action.name, device.name) meth(device) def _cells_for_device(self, dev): dhcp = [] if dev.dhcp4: dhcp.append('v4') if dev.dhcp6: dhcp.append('v6') if dhcp: dhcp = ",".join(dhcp) else: dhcp = '-' addresses = [] for v in 4, 6: if dev.configured_ip_addresses_for_version(v): addresses.extend([ "{} (static)".format(a) for a in dev.configured_ip_addresses_for_version(v) ]) elif dev.dhcp_for_version(v): if v == 4: fam = AF_INET elif v == 6: fam = AF_INET6 for a in dev._net_info.addresses.values(): log.debug("a %s", a.serialize()) if a.family == fam and a.source == 'dhcp': addresses.append("{} (from dhcp)".format( a.address)) if addresses: addresses = ", ".join(addresses) else: addresses = '-' return (dev.name, dev.type, dhcp, addresses) def new_link(self, new_dev): for i, cur_dev in enumerate(self.cur_netdevs): if cur_dev.name > new_dev.name: netdev_i = i break else: netdev_i = len(self.cur_netdevs) new_rows = self._rows_for_device(new_dev, netdev_i) self.device_table.insert_rows(3*netdev_i+1, new_rows) def update_link(self, dev): row = self.dev_to_row[dev] for i, text in enumerate(self._cells_for_device(dev)): row.columns[2*(i+1)].set_text(text) def del_link(self, dev): log.debug("del_link %s", (dev in self.cur_netdevs)) if dev in self.cur_netdevs: netdev_i = self.cur_netdevs.index(dev) self.device_table.remove_rows(3*netdev_i, 3*(netdev_i+1)) del self.cur_netdevs[netdev_i] if isinstance(self._w, StretchyOverlay): stretchy = self._w.stretchy if getattr(stretchy, 'device', None) is dev: self.remove_overlay() def _rows_for_device(self, dev, netdev_i=None): if netdev_i is None: netdev_i = len(self.cur_netdevs) rows = [] name, typ, dhcp, addresses = self._cells_for_device(dev) actions = [] for action in NetDevAction: meth = getattr(self, '_action_' + action.name) opens_dialog = getattr(meth, 'opens_dialog', False) if dev.supports_action(action): actions.append( (_(action.value), True, (action, meth), opens_dialog)) menu = ActionMenu(actions) connect_signal(menu, 'action', self._action, dev) row = make_action_menu_row([ Text("["), Text(name), Text(typ), Text(dhcp), Text(addresses, wrap='clip'), menu, Text("]"), ], menu) self.dev_to_row[dev] = row.base_widget self.cur_netdevs[netdev_i:netdev_i] = [dev] rows.append(row) if dev.type == "vlan": info = _("VLAN {id} on interface {link}").format( **dev._configuration) else: info = " / ".join([dev.hwaddr, dev.vendor, dev.model]) rows.append(Color.info_minor(TableRow([ Text(""), (4, Text(info)), Text("")]))) rows.append(Color.info_minor(TableRow([(4, Text(""))]))) return rows def _build_model_inputs(self): netdevs = self.model.get_all_netdevs() masters = [] for master in netdevs: if not master._net_info.bond['is_master']: continue masters.append(( _("Set master to %s") % master.name, True, {'action': 'add_master', 'master': master}, False)) rows = [] rows.append(TableRow([ Color.info_minor(Text(header)) for header in ["", "NAME", "TYPE", "DHCP", "ADDRESSES", ""]])) for dev in netdevs: rows.extend(self._rows_for_device(dev)) return rows def show_network_error(self, action, info=None): self.error_showing = True self.bottom.contents[0:0] = [ (Text(""), self.bottom.options()), (Color.info_error(self.error), self.bottom.options()), ] if action == 'stop-networkd': exc = info[0] self.error.set_text( "Stopping systemd-networkd-failed: %r" % (exc.stderr,)) elif action == 'apply': self.error.set_text("Network configuration could not be applied; " "please verify your settings.") elif action == 'timeout': self.error.set_text("Network configuration timed out; " "please verify your settings.") elif action == 'down': self.error.set_text("Downing network interfaces failed.") elif action == 'canceled': self.error.set_text("Network configuration canceled.") elif action == 'add-vlan': self.error.set_text("Failed to add a VLAN tag.") elif action == 'rm-dev': self.error.set_text("Failed to delete a virtual interface.") else: self.error.set_text("An unexpected error has occurred; " "please verify your settings.") def done(self, result): if self.error_showing: self.bottom.contents[0:2] = [] self.controller.network_finish(self.model.render()) def cancel(self, button=None): self.controller.cancel()