o
K��^�B � @ s� d dl Z d dlmZ d dlZd dlZd dlZd dlmZmZ d dl Z d dl
m Z m Z d dl
mZ g d�ZddgZe�d �Zd
d � ZG d d
� d
e j�ZG dd� de j�Ze jdd�G dd� d��Ze jdd�G dd� d��Ze jdd�G dd� d��Ze jdd�G dd� d��Ze jdd�G dd� d��Z e jdd�G d d � d ��Z e jdd�G d d � d ��Z G d d!� d!�Z G d"d#� d#e �Z!G d$d%� d%e �Z"dS )&� N)�pgettext)�AF_INET�AF_INET6)�List�Optional)�netplan)�lo�bridge�tun�tap�dummy�sit�canz???�vlan�bondz subiquitycore.models.networkc C s t �| �jS �N)� ipaddress� ip_interface�version)�ip� r �4/usr/share/subiquity/subiquitycore/models/network.py� addr_version$ s r c @ sZ e Zd Zed d�Zed d�Zed d�Zed d�Zed d�Zed d�Z ed d�Z
dd � Z d
S ) � NetDevAction�Infoz Edit Wifiz Edit IPv4z Edit IPv6z Edit bondzAdd a VLAN tag�Deletec C s t t| �j| j�S r )r �type�__name__�value��selfr r r �str2 s zNetDevAction.strN) r �
__module__� __qualname__r �INFO� EDIT_WLAN� EDIT_IPV4� EDIT_IPV6� EDIT_BOND�ADD_VLAN�DELETEr! r r r r r ( s
r c @ s, e Zd Ze�� Ze�� Ze�� Ze�� ZdS )� DHCPStateN) r r" r# �enum�auto�PENDING� TIMED_OUT� RECONFIGURE�
CONFIGUREDr r r r r+ 6 s
r+ T)� auto_attribsc @ s. e Zd ZU eed< ee ed< ee ed< dS )�
DHCPStatus�enabled�state� addressesN) r r" r# �bool�__annotations__r r+ r r! r r r r r3 = �
r3 c @ s` e Zd ZU e�e�Zee e d< dZ
e e e d< e�e�Z ee e d< e�e�Z
ee e d< dS )� StaticConfigr6 N�gateway� nameservers�
searchdomains)r r" r# �attr�Factory�listr6 r r! r8 r; r r< r= r r r r r: D s
r: c @ s e Zd ZU eed< eed< dS )�
VLANConfig�id�linkN)r r" r# �intr8 r! r r r r rA L s
rA c @ s& e Zd ZU ee ed< ee ed< dS )�
WLANConfig�ssid�pskN)r r" r# r r! r8 r r r r rE R s
rE c @ s. e Zd ZU eed< ee ed< ee ed< dS )�
WLANStatus�config�
scan_state�
visible_ssidsN)r r" r# rE r8 r r! r r r r r rH X r9 rH c @ sJ e Zd ZU ee ed< eed< dZee ed< dZee ed< dd� Z dS )�
BondConfig�
interfaces�modeN�xmit_hash_policy� lacp_ratec C sD | j }d| j i}|tjv r| j|d< |tjv r | j|d< | j|d�S )NrN ztransmit-hash-policy� lacp-rate)rM �
parameters)rN �BondParameters�supports_xmit_hash_policyrO �supports_lacp_raterP rM )r rN �paramsr r r � to_configf s �
�zBondConfig.to_config)
r r" r# r r! r8 rO r rP rW r r r r rL _ s
rL c @ s� e Zd ZU dZeed< eed< eed< ee ed< eed< ee ed< ee ed< ee ed < ee ed
< eed < eed < ee ed
< ee ed< ee
ed< e ed< e ed< e ed< e ed< e
e ed< dS )�
NetDevInfoz@All the information about a NetworkDev that the view code needs.�namer � is_connected� bond_master�is_used�disabled_reason�hwaddr�vendor�model�
is_virtual�
has_configr r �wlan�dhcp4�dhcp6�static4�static6�enabled_actionsN)r r" r# �__doc__r! r8 r7 r rA rL rH r3 r: r r r r r r rX u s*
rX c @ s2 e Zd Zg d�Zh d�Zg d�ZdhZddgZdS )rS )z
balance-rrz
active-backup� balance-xor� broadcast�802.3ad� balance-tlbz balance-alb> rl rm rj )�layer2zlayer2+3zlayer3+4zencap2+3zencap3+4rl �slow�fastN)r r" r# �modesrT �xmit_hash_policiesrU �
lacp_ratesr r r r rS � s
��rS c @ s� e Zd Zdd� Zdefdd�Zdd� Zdd � Zd
d � Zd d
� Z e
dd� �Z e j dd� �Z dd� Z
e
dd� �Zdd� Ze
dd� �Ze
dd� �Ze
dd � �Ze
d d � �Ze
d d � �Zd!Ze
d"d#� �Zd!Zd!Ze
d$d#� �Ze
d%d#� �Ze
d&d#� �Zd'd(� Z d)S )*�
NetworkDevc C s: || _ || _|| _i | _d | _d | _i | _d d d�| _d S )N�� � )�_model�_namer rI �infor] � dhcp_events� _dhcp_state)r r` rY �typr r r �__init__� s �zNetworkDev.__init__�returnc s� � j dkr t� jj�}nd}d }� j�� D ]}|j dkr q� j|j�dg �v r,|j} nq� j dkrP� jd urP� jd }t � jd |d |�d�|�d�d �}nd }� j d
kri� jd urit
� jd � jd d
�}nd }� j dkr�� j \}}t t
||d�� jjd � jjd d�} nd } � �� }
g g d�} � jd ur�� j�dg �D ] } | t| � �| � q�� j�di �}
ni }
i }i }dD ]9}t� �|�� j| |
| d�||< � jd ur߈ j�dt|� �}nd }t| | ||
�dg �|
�dg �d�||< q�td.i d� j�d� j �d |�d
|�d |�d|�d| �d |d �d |d! �d"|d �d#|d! �d$� j�d%� j�d&� fd'd(�tD ��d)t� jd)d ��d*t� jd*d ��d+t� jd+d ��d,� j �d-� jd u��S )/N�ethTr rM rR rN zxmit-hash-policyrQ )rM rN rO rP r rB rC )rB rC rc )rF rG rJ rK )rI rJ rK ru r6 r< )r4 r5 r6 r; �search)r6 r; r< r= rY r rZ r[ rd rv re rw rf rg r\ r] rh c s g | ] }� � |�r|�qS r )�supports_action)�.0�actionr r r �
<listcomp> s ��z*NetworkDev.netdev_info.<locals>.<listcomp>r^ r_ r` ra rb r ) r r7 rz rZ rx �get_all_netdevsrY rI �getrL rA �configured_ssidrH rE rc �dhcp_addressesr �appendr3 � dhcp_enabledr| r! r: rX r\ r] r �getattrra )r rZ r[ �dev2rV r r rF rG rc r� �configured_addresseses�addr�ns�
dhcp_statuses�static_configs�vr; r r r � netdev_info� s�
�
�
�
�
��������� �
� � �
�
������
�zNetworkDev.netdev_infoc C sj g g d�}| j d ur3| j j�� D ]"}|jtkrd}n |jtkr"d}nq|jdkr2|| �t|j �� q|S )Nru rv rw �dhcp)
rz r6 �values�familyr r �sourcer� r! �address)r �r�ar� r r r r� s
�zNetworkDev.dhcp_addressesc C s$ | j d u rdS | j �dj|d�d�S )NF�dhcp{v}�r� )rI r� �format�r r r r r r� & s
zNetworkDev.dhcp_enabledc C s$ | j �dj|d�d�s
d S | j| S )Nr� r� F)rI r� r� r| r� r r r �
dhcp_state, s
zNetworkDev.dhcp_statec C s || j |< d S r )r| )r r r5 r r r �set_dhcp_state1 s zNetworkDev.set_dhcp_statec C s | j S r )ry r r r r rY 4 s zNetworkDev.namec C s� || j kr=| jr=|| jjv rtdj| j |d���| | jj|< | jd ur=t| j| j | j�}|| jj| j < d |_ | j|_d | _|| _
d S )Nz#renaming {old_name} over {new_name})�old_name�new_name) rY ra rx �devices_by_name� RuntimeErrorr� rz rt r rI ry )r r� � dead_devicer r r rY 8 s ��
c C s t | d|j �S )N�
_supports_)r� rY )r r� r r r r� K s zNetworkDev.supports_actionc C s4 | j �di ��� D ]\}}|�d�}||f S dS )N�
access-points�password)NN)rI r� �items)r rF �settingsrG r r r r� N s
zNetworkDev.configured_ssidc C sF | j �di �}|�� |d ur i ||< |d ur!||| d< d S d S d S )Nr� r� )rI �
setdefault�clear)r rF rG �apsr r r � set_ssid_pskU s �zNetworkDev.set_ssid_pskc C s | j d ur | j jS d S r )rz �ifindexr r r r r� ] s
zNetworkDev.ifindexc C s
| j tv S r )r �"NETDEV_ALLOWED_VIRTUAL_IFACE_TYPESr r r r ra d s
zNetworkDev.is_virtualc C s8 | j �� D ]}|jdkr| j|j�dg �v r dS qdS )Nr rM TF�rx r� r rY rI r� �r �devr r r �
is_bond_slaveh s
�zNetworkDev.is_bond_slavec C sZ | j �� D ]%}|jdkr| j|j�dg �v r dS |jdkr*| j|j�d�kr* dS qdS )Nr rM Tr rC Fr� r� r r r r\ p s
�zNetworkDev.is_usedc C s dd� t | jj�� �D �S )Nc S s g | ] \}}|j d kr|j�qS )�global)�scoper )r� �_r� r r r r� } s
�z9NetworkDev.actual_global_ip_addresses.<locals>.<listcomp>)�sortedrz r6 r� r r r r �actual_global_ip_addresses{ s z%NetworkDev.actual_global_ip_addressesTc C �
| j dkS )Nrc �r r r r r �<lambda>� �
zNetworkDev.<lambda>c C r� �Nr r� r r r r r� � r� c C s | j dko| j S )Nr )r r� r r r r r� � s c C s | j o| j S r )ra r\ r r r r r� � s c C s~ | j �dj|d�d � | j �dj|d�d � g }| j �dg �D ]
}t|�|kr,|�|� q |r6|| j d< d S | j �dd � d S )Nr� r� z
gateway{v}r6 )rI �popr� r� r r� )r r �addrsr r r r � remove_ip_networks_for_version� s
�z)NetworkDev.remove_ip_networks_for_versionN) r r" r# r~ rX r� r� r� r� r� �propertyrY �setterr� r� r� r� ra r� r\ r� �_supports_INFO�_supports_EDIT_WLAN�_supports_EDIT_IPV4�_supports_EDIT_IPV6�_supports_EDIT_BOND�_supports_ADD_VLAN�_supports_DELETEr� r r r r rt � sJ
R
�� rt c @ sr e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
d dd�Z dd� Z dd� Z
dd� Zdd� ZdS ) � NetworkModel� c C s i | _ d| _|| _d S )NF)r� � has_network�project)r r� r r r r~ � s
zNetworkModel.__init__c C s t �� | _| j�|� d S r )r �ConfigrI �load_from_root)r � netplan_rootr r r �parse_netplan_configs� s
z"NetworkModel.parse_netplan_configsc C s� t �d||j|j� |jtv rt �d� d S |j}|jdkr d}|r.|jtvr.t �d� d S | j�|j�}|d urF|j d urA |S ||_ |S | j
� |�}|rW|sWt �d� d S t | |j|j�}||_ ||_
t �d||jt
�|j
�� || j|j< |S )Nznew_link %s %s %szignoring based on typerc Fz ignoring based on is_virtualz&ignoring virtual device with no configz new_link %s %s with config %s)�log�debugrY r �NETDEV_IGNORED_IFACE_TYPESra r� r� r� rz rI �config_for_devicert r �sanitize_interface_config)r r� rC ra r� rI r r r �new_link� s<
� �
� zNetworkModel.new_linkc C s* | j �� D ]
\}}|j|kr| S qd S r )r� r� r� �r r� rY r� r r r � update_link� s
��zNetworkModel.update_linkc C sX | j �� D ]$\}}|j|kr)d |_|jr!|jd u r | j |= | S | j |= | S qd S r )r� r� r� rz ra rI r� r r r �del_link� s
���zNetworkModel.del_linkc C s4 dj ||d�}t| |d� }| j|< ||d�|_|S )Nz {name}.{tag})rY �tagr )rC rB )r� rt r� rI )r � device_namer� rY r� r r r �new_vlan� s �zNetworkModel.new_vlanc C s$ t | |d� }| j|< |�� |_|S r� )rt r� rW rI )r rY � bond_configr� r r r �new_bond� s
zNetworkModel.new_bondFc C s. dd� t | j�� �D �}|sdd� |D �}|S )Nc S s g | ]\}}|�qS r r )r� �kr� r r r r� � s z0NetworkModel.get_all_netdevs.<locals>.<listcomp>c S s g | ] }|j d ur|�qS r )rI )r� r� r r r r� � s )r� r� r� )r �include_deleted�devsr r r r� � s z NetworkModel.get_all_netdevsc C s
| j | S r )r� )r rY r r r �get_netdev_by_name� s
z NetworkModel.get_netdev_by_namec C s d� d�| j�tj|dd�g�S )N�
z,# This is the network config written by '{}'F)�default_flow_style)�joinr� r� �yaml�dump)r rI r r r �stringify_config� s � �z NetworkModel.stringify_configc C s^ dddii}ddddd�}| � � D ]}||j }|d �|i �}|js&|jr,|j||j< q|S ) N�networkr � � ethernets�bonds�wifis�vlans)r� r rc r )r� r r� rI r\ rY )r rI � type_to_keyr� �key�configsr r r �
render_config� s ���
�zNetworkModel.render_configc C s$ dd| � | �� �d�ddd�d�iS )N� write_filesz$etc/netplan/00-installer-config.yaml)�path�contentz@etc/cloud/cloud.cfg.d/subiquity-disable-cloudinit-networking.cfgz network: {config: disabled}
)�etc_netplan_installer�nonet)r� r� r r r r �render s ����zNetworkModel.renderN)F)r r" r# ri r~ r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s &
r� )#r, �gettextr r �loggingr� �socketr r r> �typingr r �
subiquitycorer r� r� � getLoggerr� r �Enumr r+ �sr3 r: rA rE rH rL rX rS �objectrt r� r r r r �<module> sB
& a