o
K��^�Z � @ s8 d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlZd dlm Z m
Z
d dl m Z d dl
mZ d dlmZ d dlmZmZmZmZmZ d dlmZ d d lmZ d d
lm Z d d l m Z d d l m Z d d
l!m"Z" d dl#m$Z$m%Z% e�&d�Z'G dd� de
�Z(dZ)G dd� de�Z*G dd� d�Z+G dd� de*e e+�Z,dS )� N)�Optional)�IFF_UP�NetworkEventReceiver)�SingleInstanceTask)� with_context)�
write_file)�
BondConfig� DHCPState� NetDevAction� StaticConfig�
WLANConfig)�netplan)�BaseController)� CoreChannels)�
TuiController)�StretchyOverlay)� NetworkView)� arun_command� run_commandz subiquitycore.controller.networkc s< e Zd Zdd� Zdd� Zdd� Zdd� Z� fd d
�Z� ZS ) � SubiquityNetworkEventReceiverc C s || _ |j| _t� | _d S �N)�
controller�model�set�default_routes)�selfr � r �9/usr/share/subiquity/subiquitycore/controllers/network.py�__init__6 s z&SubiquityNetworkEventReceiver.__init__c C s* | j �||�}|d ur| j�|� d S d S r )r �new_linkr )r �ifindex�link�netdevr r r r ; s �z&SubiquityNetworkEventReceiver.new_linkc C sL | j �|�}|| jv r| j�|� | j�| j� |d ur$| j�|� d S d S r )r �del_linkr �remover �update_default_routes)r r r" r r r r# @ s
�z&SubiquityNetworkEventReceiver.del_linkc C sb | j �|�}|d u r d S t|jdd�}|t@ s)|| jv r)| j�|� | j�| j� | j�|� d S )N�flagsr ) r � update_link�getattr�infor r r$ r r% )r r r" r& r r r r' H s z)SubiquityNetworkEventReceiver.update_linkc s� t � �||� |d dkrd S |d dkrd S |d }|dks#|dkr*| j�|� n|dkr9|| jv r9| j�|� t�d | j� | j�| j� d S )
N�dst�default�table� r �NEW�CHANGE�DELzdefault routes %s) �super� route_changer �addr$ �log�debugr r% )r �action�datar �� __class__r r r2 R s z*SubiquityNetworkEventReceiver.route_change) �__name__�
__module__� __qualname__r r r# r' r2 �
__classcell__r r r8 r r 5 s
r a�
network:
version: 2
ethernets:
"all-en":
match:
name: "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
"all-eth":
match:
name: "eth*"
dhcp4: true
wifis:
"wlsp4":
dhcp4: true
access-points:
"some-ap":
password: password
c s� e Zd ZdZdZ� fdd�Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z e dd� �Z
dLdd�Zdd� Zdd� Zdd � Zed d d d �d!d"� �Zd#ed$ed%ed&dfd'd(�Zd#ed$ed&dfd)d*�Zd#ed$ed&dfd+d,�Zd#ed-efd.d/�Zd#efd0d1�Zd2e e d3ed4e d&dfd5d6�Z d#ed&efd7d8�Z d#ed9e d&dfd:d;�Z!d#ed&dfd<d=�Z"e#j$d>d?� �Z%e#j$d@dA� �Z&e#j$dBdC� �Z'e#j$dDdE� �Z(e#j$dFdG� �Z)e#j$dHdI� �Z*e#j$dJdK� �Z+� Z,S )M�BaseNetworkController�network�/c s� t � �|� t| j�| _| jjrPtj� | jj
�| _ | j }tj�
|�}tj�|�r1dd l}|�|� t�|� t|d��
}|�t� W d � n1 sKw Y | �� d| _t| �| _d S )Nr �wF)r1 r r �
_apply_config�apply_config_task�opts�dry_run�os�path�abspath� output_base�root� netplan_path�dirname�exists�shutil�rmtree�makedirs�open�write�default_netplan�parse_netplan_configs� _watchingr �network_event_receiver)r �apprK � netplan_dirrN �fpr8 r r r � s
�z BaseNetworkController.__init__c C s | j �| j� d S r )r rT rJ �r r r r rT � s z+BaseNetworkController.parse_netplan_configsc C s* g | _ | jj�| j�\| _| _| �� d S r )�_observer_handlesrW �prober�
probe_networkrV �observer�
_observer_fds�start_watchingrZ r r r �start� s
� zBaseNetworkController.startc C s2 | j sd S t�� }| jD ]}|�|� q d| _ d S �NF)rU �asyncio�get_event_loopr_ �
remove_reader�r �loop�fdr r r �
stop_watching� s
z#BaseNetworkController.stop_watchingc C s8 | j rd S t�� }| jD ]
}|�|| j|� q d| _ d S )NT)rU rc rd r_ �
add_reader� _data_readyrf r r r r` � s
z$BaseNetworkController.start_watchingc C sR t g d��}|jdkr!t�d� | �� t�� }|�d| j� d S | j �
|� d S )N)�udevadm�settlez-t�0r z*waiting 0.1 to let udev event queue settle皙�����?) r �
returncoder4 r5 ri rc rd �
call_laterr` r^ �
data_ready)r rh �cprg r r r rk � s
z!BaseNetworkController._data_readyc C s� t �d� | j�� D ]7}d}|jd u s|jsq
|jj�� D ] }|jdkr(d} nq |sA|� d� |� d� t �d|j
� t d�|_ q
d S ) Nzupdating initial NIC configF�globalT� � z disabling %szautoconfiguration failed)
r4 r5 r �get_all_netdevsr) �config� addresses�values�scope� remove_ip_networks_for_version�name�_�disabled_reason)r �dev�has_global_address�ar r r �update_initial_configs� s"
�
��z,BaseNetworkController.update_initial_configsc C s( | j jdkr d}nd}tj�| jd|�S )N� subiquityz00-installer-config.yamlz00-snapd-config.yamlz etc/netplan)rD �projectrF rG �joinrJ )r �netplan_config_file_namer r r rK � s z"BaseNetworkController.netplan_pathNFc C s | j j||d� d S )N)�context�silent)rC �
start_sync)r r� r� r r r � apply_config� s z"BaseNetworkController.apply_configc � sT �|D ]$}zt �d|j� | jj�|jt� W q ty' t � d|j� Y qw d S )Nz
downing %sz unset_link_flags failed for %s)
r4 r5 r} r^ �
rtlistener�unset_link_flagsr r � RuntimeError� exception)r �devsr� r r r �
_down_devs� s � ��z BaseNetworkController._down_devsc � s| �|D ]8}t �d|j� dddd|jg}z t|dd�I d H W q tjy; } zt �d|j|j� W Y d }~qd }~ww d S ) Nz deleting %s�ipr! �deleter� T��checkzdeleting %s failed with %r)r4 r5 r} r �
subprocess�CalledProcessErrorr) �stderr)r r� r� �cmdrs r r r � _delete_devs� s � ���z"BaseNetworkController._delete_devsc C s� | j �� }t�dtjt�|�dd�� tj| j dd�D ]}|| j
kr#qt � ||d | j
j � qt| j
| j �|�� | �� d S )Nznetwork config:
%sF��default_flow_styleT)�maskedz.dist-)r �
render_configr4 r5 �yaml�dumpr
�sanitize_config�configs_in_rootrJ rK rF �renamerD r� r �stringify_configrT )r rx �pr r r �
_write_config� s
��
z#BaseNetworkController._write_configr� zsilent={silent}�INFO)r} � description�levelc � s� �g }g }g }t � }� jjdd�D ]T}i |_dD ]-}|�|�rE�s0|�|tj� � j� |j
� n|�|tj � t �
� |j|< } |�| � q|jd u rLq|j� jj�|j�kre|jr`|�|� q|�|� q� �� �sp� �� z�� �fdd�}
� jjr�d� jj } tdt| �g�I d H t j � d�r�td d
d � j gdd �I d H nz|s�|r�ztg d
�dd �I d H tg d�dd �I d H W n
t j!y� |
d� � w |rֈ �"|�I d H |r� �#|�I d H |s�|r�tg d�dd �I d H z
td dgdd �I d H W n t j!�y
|
d� � w |�s|�r tg d�dd �I d H W ��s$� �$� n ��s-� �$� w w |�s3d S zt �%t �&dd� |D ��d�I d H W n t j'�yQ Y nw |D ] \}}i |_|�(� | �sp|�|tj)� � j� |j
� �qTd S )NT)�include_deleted)ru rv c s �s � � | � d S d S r )� apply_error)�stage�r r� r r �error s �z2BaseNetworkController._apply_config.<locals>.error� �sleepz/lib/netplan/generater
�generatez--rootr� )� systemctl�mask� --runtime�systemd-networkd.service�systemd-networkd.socket)r� �stopr� r� z
stop-networkd)r� �unmaskr� r� r� �apply)r� ra r� Fc S s h | ]}|� � �qS r )�wait)�.0�er r r � <setcomp>W s z6BaseNetworkController._apply_config.<locals>.<setcomp>�
)*r r rw � dhcp_events� dhcp_enabled�set_dhcp_stater �PENDINGrV r' r � RECONFIGURErc �Eventr3 r) rx �config_for_device�
is_virtual�appendr� �apply_startingrD rE rW � scale_factorr �strrF rG rM rJ r� r� r� r� �apply_stopping�wait_forr� � TimeoutError�dhcp_addresses� TIMED_OUT) r r� r� �devs_to_delete� devs_to_down�dhcp_device_versionsr� r� �vr� r� �delayr r� r rB � s� �
�
�
� �� ��� �� ����
��� ��z#BaseNetworkController._apply_config�dev_name�
ip_version�
static_config�returnc C s� | j �|�}|�|� |j�dg ��|j� dj|d�}|jr&|j|j|< n|j� |d � |j�di �}|�dg ��|j
� |�dg ��|j � | � |� | �
� d S )Nry z
gateway{v}�r� � nameservers�search)r �get_netdev_by_namer| rx �
setdefault�extendry �format�gateway�popr� �
searchdomainsr' r� )r r� r� r� r� �gwkey�nsr r r �set_static_configb s
z'BaseNetworkController.set_static_configc C sB | j �|�}|�|� dj|d�}d|j|< | �|� | �� d S )Nzdhcp{v}r� T)r r� r| r� rx r' r� )r r� r� r� �dhcpkeyr r r � enable_dhcpr s
z!BaseNetworkController.enable_dhcpc C s, | j �|�}|�|� | �|� | �� d S r )r r� r| r' r� )r r� r� r� r r r �disable_networkz s
z%BaseNetworkController.disable_network�idc C s: | j �||�}| �|� | j �|�}| �|� | �� d S r )r �new_vlanr r� r' r� )r r� r� �newr� r r r �add_vlan� s
z BaseNetworkController.add_vlanc C s� | j �|�}t� }|jdkr"|jd D ]
}| j �|�}|�|� qn|jdkr5| j �|jd �}|�|� d |_| �|� |D ]}| �|� q?| �� d S )N�bond�
interfaces�vlanr! ) r r� r �typerx r3 r# r' r� )r r� r� �touched_devices� device_name� interfacer! r r r � delete_link� s
�
z!BaseNetworkController.delete_link�
existing_name�new_name�new_infoc
C s� | j j}t� }|jD ]}||�}i |_|�|� q
|d u r*| j �||�}| �|� n<||�} | jd D ] }
|�||
�� q3| j�|� � � | j
|kra| j} d | _| � | � | | _|| _
| �| � n|�| � |D ]} | � | � qh| �
� d S )Nr� )r r� r r� rx r3 �new_bondr �update� to_configr} r# r' r� )
r r� r� r� r� r� r� �device�new_dev�existingr� rx r� r r r �add_or_update_bond� s0
z(BaseNetworkController.add_or_update_bondc � s: �| j �|�}|jd urtj|j�� dd�S dj|jd�S )NFr� z0Configured but not yet created {type} interface.)r� )r r� r) r� r� � serializer� r� )r r� r� r r r �get_info_for_netdev� s �
��z)BaseNetworkController.get_info_for_netdev�wlanc C sP | j �|�}|j\}}|jr|sd|jd< |�|j|j� | �|� | �� d S )NT�dhcp4) r r� �configured_ssid�ssidrx � set_ssid_psk�pskr' r� )r r� r r� �cur_ssid�cur_pskr r r �set_wlan� s
z BaseNetworkController.set_wlanc
C sd | j �|�}z | j�|j� W n ty* } zd|f |jjd< W Y d }~nd }~ww | �|� d S )Nzerror %s�
scan_state) r r� r^ � trigger_scanr r� r) r r' )r r� r� �rr r r �
start_scan� s ��z BaseNetworkController.start_scanc C � d S r r rZ r r r r� � � z$BaseNetworkController.apply_startingc C r r r rZ r r r r� � r z$BaseNetworkController.apply_stoppingc C r r r �r r� r r r r� � r z!BaseNetworkController.apply_errorc C s |r | j j�tj� d S d S r )rW �hub� broadcastr �
NETWORK_UP�r �routesr r r r% � s �z+BaseNetworkController.update_default_routesc C r r r �r r" r r r r � r z BaseNetworkController.new_linkc C s: |j �� D ]\}}|�� | r|�|tj� |�� qd S r )r� �itemsr� r� r �
CONFIGUREDr )r r" r� r� r r r r' � s �z!BaseNetworkController.update_linkc C r r r r r r r r# � r z BaseNetworkController.del_linkrb )-r: r; r< �
model_namerJ r rT ra ri r` rk r� �propertyrK r� r� r� r� r rB r� �intr r� r� r� r� r� r r r� r r r r
�abc�abstractmethodr� r� r� r% r r' r# r= r r r8 r r> sj
�
e�
�
��
�
r> c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�NetworkAnswersMixinc � sV �| j �dd�r| �� d S | j �dd�r)| j d }| j �� | �|�I d H d S d S )Nzaccept-defaultF�actions)�answers�get�done�clear� _run_actions)r r r r r � run_answers� s �
�z NetworkAnswersMixin.run_answersc C sh |d � � }|d dkr-|d dkr | jjt|d � }n
|d dkr'|d }| jj| S td�|���)Nr r� r� �index� r} zcould not resolve {})�split�view�cur_netdev_namesr �dev_name_to_table� Exceptionr� )r r� �dev_specr} r r r � _action_get� s z NetworkAnswersMixin._action_getc s" � fdd�|D �}t �d|� |S )Nc s g | ] }� � |�jj�qS r )r. �dev_infor} )r� r� rZ r r �
<listcomp> s z@NetworkAnswersMixin._action_clean_interfaces.<locals>.<listcomp>z%s)r4 r5 )r �devicesr r rZ r �_action_clean_interfaces s z,NetworkAnswersMixin._action_clean_interfacesc C sr �t �d|� d|v r�| �|d �}t| jjd�|d ��}tt|d �}| jj�d ||f|� d V | jjj }|d dkrjd}|j
j | j j
v rdt�d�I d H |d7 }|dkr\td ��|j
j | j j
v sHt �d
|� t|t�sqd S |�� D ]8\}}|�d �squd } d
}
d|v r�|�d�d } | d } | d }
| �t|j| �||�|
d��2 z3 d H W } q�6 qud S |d dk�r#| jj�� d V | jjj }|d �� }
d|
v r�|
�d�|
d< | �|jj |
|�d
d��2 z3 d H W } q�6 d}|
d | j j
v�rt�d�I d H |d7 }|dk�r td��|
d | j j
vs�|dk�r t �d|� d V d S |d dk�r2| jj� � d S td�|���)Nz_answers_action %r�objz
_action_{}r6 �DELETEg ro g @z%interface did not disappear in 5 secsz$waited %s for interface to disappearr7 �form�submit�-r �_formz-submitTz create-bondr1 r� r} z bond did not appear in 5 secsz waited %s for bond to appearr" zcould not process action {}) r4 r5 r. r( �ui�bodyr� r
�_action�_wr/ r} r) r* rc r� r, �
isinstancer r �endswithr( �_enter_form_data�stretchyr! � _create_bond�copyr� r5 r" )r r6 r, �meth�
action_objr: �t�kr� � form_name�
submit_key�prefixr~ r7 r r r �_answers_action s� � �
��
���
��
�
z#NetworkAnswersMixin._answers_actionN)r: r; r< r% r. r2 rJ r r r r r � s
r c s� e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Z� fd d �Z� fd
d�Z � fdd�Z
� fdd�Z � fdd�Z � fdd�Z
� fdd�Z� ZS )�NetworkControllerc s t � �|� d | _d| _d S rb )r1 r r) �
view_shown)r rW r8 r r r M s
zNetworkController.__init__c C s\ | j s| �� dd� | j�� D �}t| |�| _| j s#| jdd� d| _ | j�| jj � | jS )Nc S s g | ]}|� � �qS r )� netdev_info)r� r� r r r r0 U s �z-NetworkController.make_ui.<locals>.<listcomp>T)r� )
rL r� r rw r r) r� r% rV r )r � netdev_infosr r r �make_uiR s � �zNetworkController.make_uic C s
d | _ d S r )r) rZ r r r �end_ui` s
zNetworkController.end_uic C s( t �d� t| jj�| j_| j�� d S )Nz"NetworkController.done next_screen) r4 r5 �boolrV r r � has_networkrW � next_screenrZ r r r r" c s
�zNetworkController.donec C s | j �� d S r )rW � prev_screenrZ r r r �canceli s zNetworkController.cancelc �& t � �� | jd ur| j�� d S d S r )r1 r� r) �show_apply_spinnerrZ r8 r r r� l �
�z NetworkController.apply_startingc rV r )r1 r� r) �hide_apply_spinnerrZ r8 r r r� q rX z NetworkController.apply_stoppingc s* t � �|� | jd ur| j�|� d S d S r )r1 r� r) �show_network_errorr r8 r r r� v s
�z NetworkController.apply_errorc s& t � �|� | jr| j�|� d S d S r )r1 r% r) r r8 r r r% { s �z'NetworkController.update_default_routesc �. t � �|� | jd ur| j�|�� � d S d S r )r1 r r) rM r r8 r r r � �
�zNetworkController.new_linkc r[ r )r1 r' r) rM r r8 r r r' � r\ z NetworkController.update_linkc r[ r )r1 r# r) rM r r8 r r r# � r\ zNetworkController.del_link)r: r; r< r rO rP r" rU r� r� r� r% r r' r# r= r r r8 r rK J s rK )-r rc �loggingrF r� �typingr r� �probert.networkr r �subiquitycore.async_helpersr �subiquitycore.contextr �subiquitycore.file_utilr � subiquitycore.models.networkr r r
r r �
subiquitycorer
�subiquitycore.controllerr �subiquitycore.pubsubr �subiquitycore.tuicontrollerr �subiquitycore.ui.stretchyr � subiquitycore.ui.views.networkr �subiquitycore.utilsr r � getLoggerr4 r rS r> r rK r r r r �<module> s:
, uX�