461 lines
15 KiB
Python
461 lines
15 KiB
Python
# 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 <http://www.gnu.org/licenses/>.
|
|
|
|
import copy
|
|
from functools import partial
|
|
import logging
|
|
import os
|
|
import random
|
|
import select
|
|
import socket
|
|
import subprocess
|
|
|
|
import yaml
|
|
|
|
from probert.network import UdevObserver
|
|
|
|
from subiquitycore.models import NetworkModel
|
|
from subiquitycore.ui.views import (NetworkView,
|
|
NetworkSetDefaultRouteView,
|
|
NetworkBondInterfacesView,
|
|
NetworkConfigureInterfaceView,
|
|
NetworkConfigureIPv4InterfaceView,
|
|
NetworkConfigureIPv6InterfaceView,
|
|
NetworkConfigureWLANView)
|
|
from subiquitycore.ui.views.network import ApplyingConfigWidget
|
|
from subiquitycore.ui.dummy import DummyView
|
|
from subiquitycore.controller import BaseController
|
|
from subiquitycore.utils import run_command_start, run_command_summarize
|
|
|
|
log = logging.getLogger("subiquitycore.controller.network")
|
|
|
|
|
|
class BackgroundTask:
|
|
"""Something that runs without blocking the UI and can be canceled."""
|
|
|
|
def start(self):
|
|
"""Start the task.
|
|
|
|
This is called on the UI thread, so must not block.
|
|
"""
|
|
raise NotImplementedError(self.start)
|
|
|
|
def run(self):
|
|
"""Run the task.
|
|
|
|
This is called on an arbitrary thread so don't do UI stuff!
|
|
"""
|
|
raise NotImplementedError(self.run)
|
|
|
|
def end(self, observer, fut):
|
|
"""Call task_succeeded or task_failed on observer.
|
|
|
|
This is called on the UI thread.
|
|
|
|
fut is a concurrent.futures.Future holding the result of running run.
|
|
"""
|
|
raise NotImplementedError(self.end)
|
|
|
|
def cancel(self):
|
|
"""Abort the task.
|
|
|
|
Any calls to task_succeeded or task_failed on the observer will
|
|
be ignored after this point so it doesn't really matter what run
|
|
returns after this is called.
|
|
"""
|
|
raise NotImplementedError(self.cancel)
|
|
|
|
|
|
class BackgroundProcess(BackgroundTask):
|
|
|
|
def __init__(self, cmd):
|
|
self.cmd = cmd
|
|
self.proc = None
|
|
|
|
def __repr__(self):
|
|
return 'BackgroundProcess(%r)'%(self.cmd,)
|
|
|
|
def start(self):
|
|
self.proc = run_command_start(self.cmd)
|
|
|
|
def run(self):
|
|
stdout, stderr = self.proc.communicate()
|
|
return run_command_summarize(self.proc, stdout, stderr)
|
|
|
|
def end(self, observer, fut):
|
|
result = fut.result()
|
|
if result['status'] == 0:
|
|
observer.task_succeeded()
|
|
else:
|
|
observer.task_failed(result['err'])
|
|
|
|
def cancel(self):
|
|
if self.proc is None:
|
|
return
|
|
try:
|
|
self.proc.terminate()
|
|
except ProcessLookupError:
|
|
pass # It's OK if the process has already terminated.
|
|
|
|
|
|
class PythonSleep(BackgroundTask):
|
|
|
|
def __init__(self, duration):
|
|
self.duration = duration
|
|
self.r, self.w = os.pipe()
|
|
|
|
def __repr__(self):
|
|
return 'PythonSleep(%r)'%(self.duration,)
|
|
|
|
def start(self):
|
|
pass
|
|
|
|
def run(self):
|
|
r, _, _ = select.select([self.r], [], [], self.duration)
|
|
if not r:
|
|
return True
|
|
os.close(self.r)
|
|
os.close(self.w)
|
|
|
|
def end(self, observer, fut):
|
|
if fut.result():
|
|
observer.task_succeeded()
|
|
else:
|
|
observer.task_failed()
|
|
|
|
def cancel(self):
|
|
os.write(self.w, b'x')
|
|
|
|
|
|
class WaitForDefaultRouteTask(BackgroundTask):
|
|
|
|
def __init__(self, timeout, udev_observer):
|
|
self.timeout = timeout
|
|
self.udev_observer = udev_observer
|
|
|
|
def __repr__(self):
|
|
return 'WaitForDefaultRouteTask(%r)'%(self.timeout,)
|
|
|
|
def got_route(self):
|
|
os.write(self.success_w, b'x')
|
|
|
|
def start(self):
|
|
self.fail_r, self.fail_w = os.pipe()
|
|
self.success_r, self.success_w = os.pipe()
|
|
self.udev_observer.add_default_route_waiter(self.got_route)
|
|
|
|
def run(self):
|
|
try:
|
|
r, _, _ = select.select([self.fail_r, self.success_r], [], [], self.timeout)
|
|
return self.success_r in r
|
|
finally:
|
|
os.close(self.fail_r)
|
|
os.close(self.fail_w)
|
|
os.close(self.success_r)
|
|
os.close(self.success_w)
|
|
|
|
def end(self, observer, fut):
|
|
if fut.result():
|
|
observer.task_succeeded()
|
|
else:
|
|
observer.task_failed('timeout')
|
|
|
|
def cancel(self):
|
|
os.write(self.fail_w, b'x')
|
|
|
|
|
|
class TaskSequence:
|
|
def __init__(self, run_in_bg, tasks, watcher):
|
|
self.run_in_bg = run_in_bg
|
|
self.tasks = tasks
|
|
self.watcher = watcher
|
|
self.canceled = False
|
|
self.stage = None
|
|
self.curtask = None
|
|
|
|
def run(self):
|
|
self._run1()
|
|
|
|
def cancel(self):
|
|
if self.curtask is not None:
|
|
log.debug("canceling %s", self.curtask)
|
|
self.curtask.cancel()
|
|
self.canceled = True
|
|
|
|
def _run1(self):
|
|
self.stage, self.curtask = self.tasks[0]
|
|
self.tasks = self.tasks[1:]
|
|
log.debug('running %s for stage %s', self.curtask, self.stage)
|
|
self.curtask.start()
|
|
self.run_in_bg(self.curtask.run, lambda fut:self.curtask.end(self, fut))
|
|
|
|
def task_succeeded(self):
|
|
if self.canceled:
|
|
return
|
|
self.watcher.task_complete(self.stage)
|
|
if len(self.tasks) == 0:
|
|
self.watcher.tasks_finished()
|
|
else:
|
|
self._run1()
|
|
|
|
def task_failed(self, info=None):
|
|
if self.canceled:
|
|
return
|
|
self.watcher.task_error(self.stage, info)
|
|
|
|
|
|
def sanitize_config(config):
|
|
"""Return a copy of config with passwords redacted."""
|
|
config = copy.deepcopy(config)
|
|
for iface, iface_config in config.get('network', {}).get('wifis', {}).items():
|
|
for ap, ap_config in iface_config.get('access-points', {}).items():
|
|
if 'password' in ap_config:
|
|
ap_config['password'] = '<REDACTED>'
|
|
return config
|
|
|
|
class SubiquityObserver(UdevObserver):
|
|
def __init__(self, model, ui, loop):
|
|
UdevObserver.__init__(self)
|
|
self.model = model
|
|
self.ui = ui
|
|
self.loop = loop
|
|
self.default_route_waiter = None
|
|
self.default_routes = set()
|
|
|
|
def start(self):
|
|
fds = super().start()
|
|
for fd in fds:
|
|
self.loop.watch_file(fd, partial(self.data_ready, fd))
|
|
return fds
|
|
|
|
def new_link(self, ifindex, link):
|
|
self.model.new_link(ifindex, link)
|
|
|
|
def del_link(self, ifindex):
|
|
self.model.del_link(ifindex)
|
|
if ifindex in self.default_routes:
|
|
self.default_routes.remove(ifindex)
|
|
|
|
def update_link(self, ifindex):
|
|
self.model.update_link(ifindex)
|
|
|
|
def route_change(self, action, data):
|
|
if data['dst'] != b'default':
|
|
return
|
|
if data['table'] != 254:
|
|
return
|
|
super().route_change(action, data)
|
|
ifindex = data['ifindex']
|
|
if action == "NEW":
|
|
self.default_routes.add(ifindex)
|
|
if self.default_route_waiter:
|
|
self.default_route_waiter()
|
|
elif action == "DEL" and ifindex in self.default_routes:
|
|
self.default_routes.remove(ifindex)
|
|
log.debug('default routes %s', self.default_routes)
|
|
|
|
def add_default_route_waiter(self, waiter):
|
|
if self.default_routes:
|
|
waiter()
|
|
else:
|
|
self.default_route_waiter = waiter
|
|
|
|
def refresh(self):
|
|
v = self.ui.frame.body
|
|
if hasattr(v, 'refresh_model_inputs'):
|
|
v.refresh_model_inputs()
|
|
|
|
def data_ready(self, fd):
|
|
code = subprocess.call(['udevadm', 'settle', '-t', '0'])
|
|
if code != 0:
|
|
log.debug("waiting 0.1 to let udev event queue settle")
|
|
self.loop.set_alarm_in(0.1, lambda loop, ud:self.data_ready(fd))
|
|
super().data_ready(fd)
|
|
self.refresh()
|
|
|
|
|
|
default_netplan = '''
|
|
network:
|
|
version: 2
|
|
ethernets:
|
|
"en*":
|
|
addresses:
|
|
- 10.0.2.15/24
|
|
gateway4: 10.0.2.2
|
|
nameservers:
|
|
addresses:
|
|
- 8.8.8.8
|
|
- 8.4.8.4
|
|
search:
|
|
- foo
|
|
- bar
|
|
"eth*":
|
|
dhcp4: true
|
|
wifis:
|
|
"wl*":
|
|
dhcp4: true
|
|
access-points:
|
|
"some-ap":
|
|
password: password
|
|
'''
|
|
|
|
class NetworkController(BaseController):
|
|
signals = [
|
|
('menu:network:main:set-default-v4-route', 'set_default_v4_route'),
|
|
('menu:network:main:set-default-v6-route', 'set_default_v6_route'),
|
|
]
|
|
|
|
root = "/"
|
|
|
|
def __init__(self, common):
|
|
super().__init__(common)
|
|
if self.opts.dry_run:
|
|
self.root = os.path.abspath(".subiquity")
|
|
self.tried_once = False
|
|
netplan_path = self.netplan_path
|
|
netplan_dir = os.path.dirname(netplan_path)
|
|
if os.path.exists(netplan_dir):
|
|
import shutil
|
|
shutil.rmtree(netplan_dir)
|
|
os.makedirs(netplan_dir)
|
|
with open(netplan_path, 'w') as fp:
|
|
fp.write(default_netplan)
|
|
self.model = NetworkModel(self.root)
|
|
|
|
self.observer = SubiquityObserver(self.model, self.ui, self.loop)
|
|
self.observer.start()
|
|
|
|
def start_scan(self, dev):
|
|
self.observer.wlan_listener.trigger_scan(dev.ifindex)
|
|
|
|
def cancel(self):
|
|
self.signal.emit_signal('prev-screen')
|
|
|
|
def default(self):
|
|
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")
|
|
self.ui.set_header(title, excerpt)
|
|
self.ui.set_footer(footer, 20)
|
|
self.ui.set_body(NetworkView(self.model, self))
|
|
|
|
@property
|
|
def netplan_path(self):
|
|
if self.opts.project == "subiquity":
|
|
netplan_config_file_name = '00-installer-config.yaml'
|
|
else:
|
|
netplan_config_file_name = '00-snapd-config.yaml'
|
|
return os.path.join(self.root, 'etc/netplan', netplan_config_file_name)
|
|
|
|
def network_finish(self, config):
|
|
log.debug("network config: \n%s", yaml.dump(sanitize_config(config), default_flow_style=False))
|
|
|
|
netplan_path = self.netplan_path
|
|
while True:
|
|
try:
|
|
tmppath = '%s.%s' % (netplan_path, random.randrange(0, 1000))
|
|
fd = os.open(tmppath, os.O_WRONLY | os.O_EXCL | os.O_CREAT, 0o0600)
|
|
except FileExistsError:
|
|
continue
|
|
else:
|
|
break
|
|
w = os.fdopen(fd, 'w')
|
|
with w:
|
|
w.write("# This is the network config written by '{}'\n".format(self.opts.project))
|
|
w.write(yaml.dump(config))
|
|
os.rename(tmppath, netplan_path)
|
|
self.model.parse_netplan_configs()
|
|
if self.opts.dry_run:
|
|
tasks = [
|
|
('one', BackgroundProcess(['sleep', '0.1'])),
|
|
('two', PythonSleep(0.1)),
|
|
('three', BackgroundProcess(['sleep', '0.1'])),
|
|
]
|
|
if os.path.exists('/lib/netplan/generate'):
|
|
# If netplan appears to be installed, run generate to at
|
|
# least test that what we wrote is acceptable to netplan.
|
|
tasks.append(('generate', BackgroundProcess(['netplan', 'generate', '--root', self.root])))
|
|
if not self.tried_once:
|
|
tasks.append(('timeout', WaitForDefaultRouteTask(3, self.observer)))
|
|
tasks.append(('fail', BackgroundProcess(['false'])))
|
|
self.tried_once = True
|
|
else:
|
|
tasks = [
|
|
('generate', BackgroundProcess(['/lib/netplan/generate'])),
|
|
('apply', BackgroundProcess(['netplan', 'apply'])),
|
|
('timeout', WaitForDefaultRouteTask(30, self.observer)),
|
|
]
|
|
|
|
def cancel():
|
|
self.cs.cancel()
|
|
self.task_error('canceled')
|
|
self.acw = ApplyingConfigWidget(len(tasks), cancel)
|
|
self.ui.frame.body.show_overlay(self.acw)
|
|
|
|
self.cs = TaskSequence(self.run_in_bg, tasks, self)
|
|
self.cs.run()
|
|
|
|
def task_complete(self, stage):
|
|
self.acw.advance()
|
|
|
|
def task_error(self, stage, info=None):
|
|
self.ui.frame.body.remove_overlay()
|
|
self.ui.frame.body.show_network_error(stage, info)
|
|
|
|
def tasks_finished(self):
|
|
self.signal.emit_signal('network-config-written', self.netplan_path)
|
|
self.signal.emit_signal('next-screen')
|
|
|
|
def set_default_v4_route(self):
|
|
self.ui.set_header("Default route")
|
|
self.ui.set_body(NetworkSetDefaultRouteView(self.model, socket.AF_INET, self))
|
|
|
|
def set_default_v6_route(self):
|
|
self.ui.set_header("Default route")
|
|
self.ui.set_body(NetworkSetDefaultRouteView(self.model, socket.AF_INET6, self))
|
|
|
|
def bond_interfaces(self):
|
|
self.ui.set_header("Bond interfaces")
|
|
self.ui.set_body(NetworkBondInterfacesView(self.model, self))
|
|
|
|
def network_configure_interface(self, iface):
|
|
self.ui.set_header("Network interface {}".format(iface))
|
|
self.ui.set_footer("")
|
|
self.ui.set_body(NetworkConfigureInterfaceView(self.model, self, iface))
|
|
|
|
def network_configure_ipv4_interface(self, iface):
|
|
self.ui.set_header("Network interface {} manual IPv4 "
|
|
"configuration".format(iface))
|
|
self.ui.set_footer("")
|
|
self.ui.set_body(NetworkConfigureIPv4InterfaceView(self.model, self, iface))
|
|
|
|
def network_configure_wlan_interface(self, iface):
|
|
self.ui.set_header("Network interface {} WIFI "
|
|
"configuration".format(iface))
|
|
self.ui.set_footer("")
|
|
self.ui.set_body(NetworkConfigureWLANView(self.model, self, iface))
|
|
|
|
def network_configure_ipv6_interface(self, iface):
|
|
self.ui.set_header("Network interface {} manual IPv6 "
|
|
"configuration".format(iface))
|
|
self.ui.set_footer("")
|
|
self.ui.set_body(NetworkConfigureIPv6InterfaceView(self.model, self, iface))
|
|
|
|
def install_network_driver(self):
|
|
self.ui.set_body(DummyView(self))
|
|
|