U
K��^�D � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d dlmZmZ d dlmZ d d lmZ d d
lmZ d d lmZmZ e� d �Z G d
d� de�Z dZ G dd� de
�Z dS )� N)�IFF_UP�NetworkEventReceiver)�SingleInstanceTask)� with_context)�BaseController)�
write_file)�BondParameters� NetDevAction)�netplan��StretchyOverlay)� NetworkView)� arun_command� run_commandz subiquitycore.controller.networkc sL e Zd Zdd� Zdd� Zdd� Zdd� Z� fd d
�Zd d � Zd
d� Z � Z
S )� SubiquityNetworkEventReceiverc C s$ || _ d | _g | _t� | _i | _d S �N)�model�view�default_route_watchers�set�default_routes� dhcp_events)�selfr � r �9/usr/share/subiquity/subiquitycore/controllers/network.py�__init__/ s
z&SubiquityNetworkEventReceiver.__init__c C s0 | j �||�}| jd k r,|d k r,| j�|� d S r )r �new_linkr )r �ifindex�link�netdevr r r r 6 s z&SubiquityNetworkEventReceiver.new_linkc C sD | j �|�}|| jkr"| j�|� | jd k r@|d k r@| j�|� d S r )r �del_linkr �remover )r r r r r r r ; s
z&SubiquityNetworkEventReceiver.del_linkc C s� | j �|�}|d krd S t|jdd�}|t@ sZ|| jkrZ| j�|� | jD ]}|| j� qJ|j� � D ] \}}|�
� | rd|� � qd| j d k r�| j �|� d S )N�flagsr )
r � update_link�getattr�infor r r! r r �items�dhcp_addressesr r )r r r r" �watcher�v�er r r r# B s
z)SubiquityNetworkEventReceiver.update_linkc s� t � �||� |d dkr d S |d dkr.d S |d }|dksF|dkrT| j�|� n |dkrr|| jkrr| j�|� | jD ]}|| j� qxt�d | j� d S )
NZdst�default�table� r ZNEWZCHANGEZDELzdefault routes %s)�super� route_changer �addr! r �log�debug)r �action�datar r( �� __class__r r r/ R s
z*SubiquityNetworkEventReceiver.route_changec C s | j �|� || j� d S r )r �appendr �r r( r r r �add_default_route_watchera s z7SubiquityNetworkEventReceiver.add_default_route_watcherc C s || j kr| j �|� d S r )r r! r8 r r r � remove_default_route_watchere s
z:SubiquityNetworkEventReceiver.remove_default_route_watcher) �__name__�
__module__� __qualname__r r r r# r/ r9 r: �
__classcell__r r r5 r r . 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 dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zed#d$� �Zd9d'd(�Zd)d*� Zd+d,� Zd-d.� Zed/d0d1d2�d3d4� �Z d5d6� Z d7d8� Z � Z S ):�NetworkControllerZnetwork�/c s� t � �|� d | _d| _t| j�| _| jjr�t j
� d�| _ | j
}t j
�|�}t j
�|�rjdd l}|�|� t �|� t|d��}|�t� W 5 Q R X | �� d| _t| j�| _| j�| j � d S )NFz
.subiquityr �w) r. r r �
view_shownr �
_apply_config�apply_config_task�opts�dry_run�os�path�abspath�root� netplan_path�dirname�exists�shutilZrmtree�makedirs�open�write�default_netplan�parse_netplan_configs� _watchingr r �network_event_receiverr9 �
route_watcher)r �apprK Z netplan_dirrN �fpr5 r r r � s(
�zNetworkController.__init__c C s | j �| j� d S r )r rS rJ �r r r r rS � s z'NetworkController.parse_netplan_configsc C s |r| j �d� d S )Nznetwork-change)�signalZ emit_signal)r Zroutesr r r rV � s z NetworkController.route_watcherc C s* g | _ | jj�| j�\| _| _| �� d S r )Z_observer_handlesrW ZproberZ
probe_networkrU �observer�
_observer_fds�start_watchingrY r r r �start� s �
zNetworkController.startc C s2 | j s
d S t�� }| jD ]}|�|� qd| _ d S )NF)rT �asyncio�get_event_loopr\ Z
remove_reader�r �loop�fdr r r �
stop_watching� s
z NetworkController.stop_watchingc C s8 | j r
d S t�� }| jD ]}|�|| j|� qd| _ d S )NT)rT r_ r` r\ Z
add_reader� _data_readyra r r r r] � s
z NetworkController.start_watchingc C s� t ddddg�}|jdkrFt�d� | �� t�� }|�d| j� d S | j �
|� | j j }t
t|dd �t�r�t|jjd �r�|jj�� d S )
NZudevadmZsettlez-t�0r z*waiting 0.1 to let udev event queue settleg�������?�_w�refresh_model_inputs)r �
returncoder1 r2 rd r_ r` Z
call_laterr] r[ Z
data_ready�ui�body�
isinstancer$ r �hasattrrg �stretchyrh )r rc �cprb r) r r r re � s
z NetworkController._data_readyc C s | j �|j� d S r )r[ Z trigger_scanr )r �devr r r �
start_scan� s z NetworkController.start_scanc C s( t �d� t| jj�| j_| j�� d S )Nz"NetworkController.done next_screen) r1 r2 �boolrU r r Z has_networkrW Z next_screenrY r r r �done� s
�zNetworkController.donec C s | j �� d S r )rW Z prev_screenrY r r r �cancel� s zNetworkController.cancelc C s� |d � � }d }|d dkr\|d dkr@| j�� t|d � }n |d dkr\| j�|d �}|d krrtd�|���t|�dkr�|d � � \}}|dkr�|�� t|� S n|S td�|���d S ) Nr Z interface� �index� �namezcould not resolve {}�part) �splitr �get_all_netdevs�intZget_netdev_by_name� Exception�format�lenZ
partitions)r �idZdev_specrp ry rv r r r � _action_get� s z NetworkController._action_getc s � fdd�|D �S )Nc s g | ]}� � |��qS r )r� )�.0�devicerY r r �
<listcomp>� s z;NetworkController._action_clean_devices.<locals>.<listcomp>r )r �devicesr rY r �_action_clean_devices� s z'NetworkController._action_clean_devicesc c sx ddl m} t�d|� d|kr�| �|d �}t| jjd�|d ��}tt |d �}| jj�
d ||f|� d V | jjj }t ||�s�d S |�
� D ]f\}}|�d�s�q�d} d }
d
|kr�|�d
�d } | d } | d }
| �t|j| �||�|
d
��E d H q�nz|d dk�rJ| jj�� d V | jjj }| �|jj|d |�d d
��E d H n*|d dk�rf| jj�� ntd�|���d S )Nr r z_answers_action %r�objz
_action_{}r3 r4 �formZsubmit�-Z_formz-submitTz create-bondrs zcould not process action {})�subiquitycore.ui.stretchyr r1 r2 r� r$ rj rk r~ r Z_actionrg rl r& �endswithrz Z_enter_form_datarn �getZ _create_bondr� rs r} ) r r3 r r� �methZ
action_objrk �kr) Z form_nameZ
submit_key�prefixr r r �_answers_action� sP �
�
� z!NetworkController._answers_actionc C s� t �d� | j�� D ]n}d}|jd ks|js.q|jj�� D ]}|jdkr:d} qRq:|s|� d� |� d� t �d|j
� t d�|_ qd S ) Nzupdating initial NIC configF�globalT� � z disabling %szautoconfiguration failed)
r1 r2 r r{ r% �configZ addresses�valuesZscopeZ remove_ip_networks_for_versionrx �_Zdisabled_reason)r rp Zhas_global_address�ar r r �update_initial_configs s
z(NetworkController.update_initial_configsc C sP | j s| �� t| j| �| _| j s4| jdd� d| _ | j| j_| j�| j� d S )NT)�silent) rB r� r
r r � apply_configrU rj Zset_bodyrY r r r �start_ui* s
zNetworkController.start_uic C s d | _ | j_ d S r )r rU rY r r r �end_ui4 s zNetworkController.end_uic C s( | j jdkrd}nd}tj�| jd|�S )NZ subiquityz00-installer-config.yamlz00-snapd-config.yamlz etc/netplan)rE �projectrG rH �joinrJ )r Znetplan_config_file_namer r r rK 7 s z NetworkController.netplan_pathNFc C s | j j||d� d S )N)�contextr� )rD Z
start_sync)r r� r� r r r r� ? s z NetworkController.apply_configc � sV |D ]L}z$t �d|j� | jj�|jt� W q tk
rN t � d|j� Y qX qd S )Nz
downing %sz unset_link_flags failed for %s)
r1 r2 rx r[ Z
rtlistenerZunset_link_flagsr r � RuntimeErrorZ exception)r �devsrp r r r �
_down_devsB s z NetworkController._down_devsc � sz |D ]p}t �d|j� dddd|jg}zt|dd�I d H W q tjk
rr } zt �d|j|j� W 5 d }~X Y qX qd S ) Nz deleting %sZipr �deleterp T�Zcheckzdeleting %s failed with %r)r1 r2 rx r �
subprocess�CalledProcessErrorr% �stderr)r r� rp �cmdro r r r � _delete_devsK s z NetworkController._delete_devsc C s� | j �� }t�dtjt�|�dd�� tj| j dd�D ](}|| j
krFq6t � ||d | j
j � q6t| j
| j �|�dd� | �� d S ) Nznetwork config:
%sF)Zdefault_flow_styleT)Zmaskedz.dist-rA )Zomode)r Z
render_configr1 r2 �yaml�dumpr
Zsanitize_configZconfigs_in_rootrJ rK rG �renamerE r� r Zstringify_configrS )r r� �pr r r �
_write_configU s"
��
�z NetworkController._write_configr� zsilent={silent}�INFO)rx Z description�levelc
� sf g }g }g }t � }� jjdd�D ]�}i |_dD ]V}|�|�r.�s\|�|d� � j�|j� n |�|d� t �
� |j|< } |� | � q.|j d kr�q |j
� jj
�|j �kr |jr�|�|� q |�|� q � �� �s� jr� j�� �z�� �fdd�}
� jj�rNd� jj } td t| �g�I d H tj � d
��rntd d d
� j gdd�I d H �n |�sZ|�r�z:tdddddgdd�I d H tddddgdd�I d H W n" t j k
�r� |
d� � Y nX |�rΈ �!|�I d H |�r� �"|�I d H |�s�|�r tdddddgdd�I d H ztd dgdd�I d H W n" t j k
�rH |
d� � Y nX |�sV|�rntdddgdd�I d H W 5 ��s�� j�r�� j�� X � j#�$dd��r�� �%� n4� j#�$dd��rڈ j#d } � j#�&� � �'� �(| �� |�s�d S z&t �)t �*d d � |D ��d �I d H W n t j+k
�r" Y nX |D ]8\}}i |_|�,� | �s(|�|d � � j�|j� �q(d S ) NT)Zinclude_deleted)r� r� ZPENDINGZ RECONFIGUREc s �s� j r� j �| � d S r )r Zshow_network_error)Zstage�r r� r r �error� s
z.NetworkController._apply_config.<locals>.errorru �sleepz/lib/netplan/generater
Zgeneratez--rootr� Z systemctl�maskz --runtimezsystemd-networkd.servicezsystemd-networkd.socket�stopz
stop-networkdZunmaskZapplyr^ Fzaccept-default�actionsc S s h | ] }|� � �qS r )�wait)r� r* r r r � <setcomp>� s z2NetworkController._apply_config.<locals>.<setcomp>�
ZTIMEDOUT)-r r r{ r Z dhcp_enabledZset_dhcp_staterU r# r r_ ZEventr0 r% r� Zconfig_for_deviceZ
is_virtualr7 r� r Zshow_apply_spinnerZhide_apply_spinnerrE rF rW Z scale_factorr �strrG rH rM rJ r� r� r� r� Zanswersr� rs �clearZ
_run_iteratorZ _run_actionsZwait_forr� � TimeoutErrorr' )
r r� r� Zdevs_to_deleteZ devs_to_downZdhcp_device_versionsr rp r) r* r� Zdelayr� r r� r rC i s�
�
� �� �� �� �
� z NetworkController._apply_configc C s | j �||�S r )r Znew_vlan)r r� Zvlanr r r �add_vlan� s zNetworkController.add_vlanc C s� |d }d|i}|t jkr&|d |d< |t jkr<|d |d< |d D ]
}i |_qDdd� |d D �}|d kr~| j�|d ||�S ||jd
< ||jd < |d |_|S d S ) N�modeZxmit_hash_policyztransmit-hash-policyZ lacp_ratez lacp-rater� c S s g | ]
}|j �qS r )rx )r� �dr r r r� � s z8NetworkController.add_or_update_bond.<locals>.<listcomp>rx �
interfacesZ
parameters)r Zsupports_xmit_hash_policyZsupports_lacp_rater� r Znew_bondrx )r �existing�resultr� Zparamsr� r� r r r �add_or_update_bond� s" �
z$NetworkController.add_or_update_bond)NF) r; r< r= Z
model_namerJ r rS rV r^ rd r] re rq rs rt r� r� r� r� r� r� �propertyrK r� r� r� r� r rC r� r� r> r r r5 r r? � s@ (
�
lr? )!r_ ZloggingrG r� r� Zprobert.networkr r Zsubiquitycore.async_helpersr Zsubiquitycore.contextr Zsubiquitycore.controllerr Zsubiquitycore.file_utilr Z subiquitycore.models.networkr r Z
subiquitycorer
r� r Z subiquitycore.ui.views.networkr
Zsubiquitycore.utilsr r Z getLoggerr1 r rR r? r r r r �<module> s$
< |