3
\�mevK � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
m Z m Z m
Z
d dlmZmZmZmZmZ d dlmZmZ dZeje�Zdd� Zde e d �d
d �Z d d
� Z ddd�Z G dd� dej �Z ddd�Z!dS )� N)�Optional�cast)�features�safeyaml�subp�util)�IPV6_DYNAMIC_TYPES�
SYS_CLASS_NET�get_devicelist�renderer�subnet_is_ipv6)�NET_CONFIG_TO_V2� NetworkStates4 # This is the initial network config.
# It can be overwritten by cloud-init or console-conf.
network:
version: 2
ethernets:
all-en:
match:
name: "en*"
dhcp4: true
all-eth:
match:
name: "eth*"
dhcp4: true
c s t � fdd�| j� D ��S )Nc 3 s$ | ] \}}|j � �r||fV qd S )N)�
startswith)�.0�key�value)�match� � /usr/lib/python3.6/netplan.py� <genexpr>* s z,_get_params_dict_by_match.<locals>.<genexpr>)�dict�items)�configr r )r r �_get_params_dict_by_match( s
r )r �entryc C sl d(dd�}|dkrg }g }g }g }g }| j dg �} | dkr>g } �x| D �]}
|
j d�} | jd�r�| dkrp| d7 } |j| d i� qF| tkr�|jd
d i� qF| d)krFd
|
j d� } d|
kr�| d|
j d� 7 } |
j d��rl|
j d�dd�}
yDtj|
d �}tj| dd�}||k�r&tjd|
d | � d |
d< W n8 t k
�r` } ztj
d|
d | |� W Y dd}~X nX |j |
� d|
k�r�|||
j dg ��7 }d|
k�r�|||
j dg ��7 }d|
k�r�d}t |
��r�d |k�r�d }|j||
j d�i� xh|
j d g �D ]X}d |j d �|j d�f }|j d�|d�}
d |k�r<|
jd |j d d!�i� |j |
� �q�W |j | � qFW d| k�r�|j d�}|�r�| d |k�r�tj
d"|| d |� n | d |d< t
|�d#k�r�|jd$|i� t
|�d#k�r�|jd |i� t
|�d#k�r d$|i}|jd%|i� t
|�d#k�r6|j d%i �}|jd&|i� |jd%|i� d'| k�rh| d' dk �rh|jd'tj| j d'��i� dS )*a� This method parse a cloudinit.net.network_state dictionary (config) and
maps netstate keys/values into a dictionary (entry) to represent
netplan yaml. (config v1 -> netplan)
An example config dictionary might look like:
{'mac_address': '52:54:00:12:34:00',
'name': 'interface0',
'subnets': [
{'address': '192.168.1.2/24',
'mtu': 1501,
'type': 'static'},
{'address': '2001:4800:78ff:1b:be76:4eff:fe06:1000",
'mtu': 1480,
'netmask': 64,
'type': 'static'}],
'type: physical',
'accept-ra': 'true'
}
An entry dictionary looks like:
{'set-name': 'interface0',
'match': {'macaddress': '52:54:00:12:34:00'},
'mtu': 1501}
After modification returns
{'set-name': 'interface0',
'match': {'macaddress': '52:54:00:12:34:00'},
'mtu': 1501,
'address': ['192.168.1.2/24', '2001:4800:78ff:1b:be76:4eff:fe06:1000"],
'ipv6-mtu': 1480}
� c S s2 | st | t� r| S || kr(| j|�S | gS dS )zT
Helper to convert strings to list of strings, handle single string
N)�
isinstance�str�split)�obj�tokenr r r �_listifyU s
z$_extract_addresses.<locals>._listifyN�subnets�typeZdhcp�4TZdhcp6�static�static6z%sZaddress�prefixz/%dZgateway�default)ZviaZtoF)�strictzAGateway %s is not contained within subnet %s, adding on-link flagzon-linkzDFailed to check whether gateway %s is contained within subnet %s: %s�dns_nameserversZ
dns_searchZmtuzipv6-mtu�routesz%s/%s�networkZmetric�d zZNetwork config: ignoring %s device-level mtu:%s because ipv4 subnet-level mtu:%s provided.r � addresses� nameservers�searchz accept-ra)r )r& r' )�getr �updater � ipaddressZ
ip_addressZ
ip_network�LOG�debug�
ValueErrorZwarning�appendr �lenr Zis_true)r r �ifnamer r" r/ r, r0 �
searchdomainsr# ZsubnetZsn_typeZaddrZ new_routeZsubnet_gatewayZsubnet_network�eZmtukeyZrouteZto_netZ entry_mtu�nsr r r �_extract_addresses0 s� %
r> c s8 t � fdd�| j� D ��}t|�dkr4|jd|i� d S )Nc s$ g | ] \}}|j d d�� kr|�qS )z bond-masterN)r2 )r �name�cfg)� bond_masterr r �
<listcomp>� s z0_extract_bond_slaves_by_name.<locals>.<listcomp>r �
interfaces)�sortedr r9 r3 )rC r rA Zbond_slave_namesr )rA r � _extract_bond_slaves_by_name� s
rE c s� t j� d�}tjj|�s d S tj|dd�}|tkr6d S � fdd�d D �}d d� |D �}tj d
||� x |g| D ]}tj
|� qpW d S ) Nz etc/netplan/00-snapd-config.yamlF)�decodec s g | ]}t j� |��qS r )r � target_path)r �f)�targetr r rB � s z"_clean_default.<locals>.<listcomp>�-run/systemd/network/10-netplan-all-en.network�.run/systemd/network/10-netplan-all-eth.network�#run/systemd/generator/netplan.stampc S s g | ]}t jj|�r|�qS r )�os�path�isfile)r rH r r r rB � s z9removing known config '%s' and derived existing files: %s)rJ rK rL ) r rG rM rN rO r Z load_file�KNOWN_SNAPD_CONFIGr5 r6 �unlink)rI Ztpath�contentZderivedZexistingrH r )rI r �_clean_default� s"
rS c @ sz e Zd ZdZddgZddgZddd�Zedd � �Zde e
e dd
�d d �Z de
e
d�dd�Zddd�Ze ed�dd�ZdS )�RendererzBRenders network information in a /etc/netplan/network.yaml format.�netplanZgenerate�infoNc C sR |si }|j dd�| _|j dd �| _|j dd�| _|j dd�| _|j dd �| _d S ) N� netplan_pathz etc/netplan/50-cloud-init.yaml�netplan_headerZpostcmdsF�
clean_defaultTr )r2 rW rX � _postcmdsrY � _features)�selfr r r r �__init__� s
zRenderer.__init__c C s� | j d kr�y0tj| jdd�\}}tj|�}|d d | _ W nF tjk
rP Y n2 ttfk
r� } ztj d|� W Y d d }~X nX | j S )NT)�capturez
netplan.ior z-Failed to list features from netplan info: %s)
r[ r � NETPLAN_INFOr Z load_yaml�ProcessExecutionError� TypeError�KeyErrorr5 r6 )r\ Z info_blobZ_errrV r< r r r r � s
zRenderer.features)�
network_state� templates�returnc
C s* t jjtj|�| j�}tjt jj|�� | j |�}| j
r>| j
nd}|j d�sT|d7 }|| }d}t jj |�r�tj
tj|j� ��}t|d��} tj
| �}
W d Q R X ||
kr�d}tjr�dnd} | r�t jj |�r�tj|�} | | @ | kr�| } tj||| d� | j�rt|d � | j| j|d
� | j| jd � d S ) N� �
F�rbTi� i� )�mode)rI )�run� same_content)rj )rM rN �joinr rG rW r Z
ensure_dir�dirname�_render_contentrX �endswith�existsZ hash_buffer�io�BytesIO�encode�openr Z NETPLAN_CONFIG_ROOT_READ_ONLYZget_permissionsZ
write_filerY rS �_netplan_generaterZ �_net_setup_link)
r\ rc rd rI ZfpnplanrR �headerrk Zhashed_contentrH Zhashed_original_contentri Z current_moder r r �render_network_state s0
z Renderer.render_network_stateF)rj rk c C s8 |st jd� d S |r$t jd� d S tj| jdd� d S )Nz!netplan generate postcmd disabledzEskipping call to `netplan generate`. reason: identical netplan configT)r^ )r5 r6 r �NETPLAN_GENERATE)r\ rj rk r r r ru 8 s
zRenderer._netplan_generatec C s� |st jd� dS dddg}t}x�td�D ]l}y>x6t� D ],}tjjt| �r8t j |t| g dd� q8W P W q* t j
k
r� } z
|}W Y dd}~X q*X q*W t d �|�dS )
z�To ensure device link properties are applied, we poke
udev to re-evaluate networkd .link files and call
the setup_link udev builtin command
z'netplan net_setup_link postcmd disabledNZudevadmz test-builtinZnet_setup_link� T)r^ zQ'udevadm test-builtin net_setup_link' unable to run successfully for all devices.) r5 r6 � Exception�ranger
rM rN �islinkr r r` � RuntimeError)r\ rj Z setup_lnkZlast_exception�_Zifacer< r r r rv D s"
zRenderer._net_setup_link)rc re c + C sd |j dkr*tjd� tjd|jiddd�S i }i }i }i }i }g }|jjdg �}|j} |j }
�x4|j
� D �]&} | jd�} t t dd � | j
� ��}
|
jd
�}|d k�r| |
jd d �d
�}|d d kr�|
jdd �}|d k r�d|j� i|d < n |d = |d= t|
|| | j� |j| |i� qh|dk�r i }i }tt tjd��}x^d/D ]V}t|
|�}xD|j
� D ]8\}}|j|jdd��}|d k�rz�qR|j||i� �qRW �q:W t|�dk�r�|jd|i� |
jd��r�|
d j� |d< |
jd�}|dk�r�t||| � t|
|| | j� |j| |i� qh|dk�r2|
jd�}ttj|��}d|i} d } t|
| �} i } tt tjd��} x�| j
� D ]t\}}| j|�}|d k�r��qb| j||i� |d0k�rbi }!x&|D ] }"|"j� \}#}$t|$�|!|#< �q�W | j||!i� �qbW t| �dk�r�| jd| i� |
jd��r|
d j� | d< t|
| | | j� |j| | i� qh|d krh|
jd �|
jd!�d"�}%|
jdd �}|d k �rr|j� |%d< t|
|%| | j� |j| |%i� qhW | �s�|
�r| |
d#�}&xV|||||gD ]D}'x<|'j
� D ]0\}(})d$|)k�s�d%|)k�r�q�|)jd$|&i� �q�W �q�W d&d'� }*|j d(� ||*d)|�7 }||*d*|�7 }||*d+|�7 }||*d,|�7 }||*d-|�7 }d.j |�S )1N� zV2 to V2 passthroughr- F)�explicit_start� explicit_endrC r? c S s | d d k S )N� r )�itr r r �<lambda>} s z*Renderer._render_content.<locals>.<lambda>r$ Zphysicalr )zset-namer Z mac_addressZ
macaddresszset-name�bond�bond_�bond-r �-r Z
parametersz bond-slavesZnone�bridge� bridge_portsZbridge_� path-cost�
port-priority�vlanZvlan_idzvlan-raw-device)�id�link)r/ r1 r0 r/ c S s0 |r,t j| |idddd�}tj|d�}|gS g S )NFT)r� r� Znoaliasr � z )r �dumps�textwrap�indent)r? �section�dumpZtxtr r r �_render_section� s z1Renderer._render_content.<locals>._render_sectionznetwork:
version: 2
� ethernets�wifis�bonds�bridges�vlansrf )r� r� )r� r� ) �versionr5 r6 r r� r Z_network_stater2 r+ Zdns_searchdomainsZiter_interfacesr �filterr �lowerr> r r3 r r
r �replacer9 rE rD �copyr �intr8 rl )+r\ rc r� r� r� r� r� rR rC r0 r; r r: ZifcfgZif_typeZethZmacaddrr� Z bond_configZ v2_bond_mapr Z bond_paramsZparamr ZnewnameZslave_interfacesr� Zportsr� Z match_prefixZparamsZ br_configZ
v2_bridge_mapZnewvalue�valZportZportvalr� Znscfgr� �_namer@ r� r r r rn c s�
zRenderer._render_content)N)NN)FF)F)�__name__�
__module__� __qualname__�__doc__ry r_ r] �propertyr r r r rx �boolru rv r rn r r r r rT � s
'
rT c C s4 dg}ddg}x |D ]}t j||| d�sdS qW dS )NrU z /usr/sbinz/sbin)r1 rI FT)r Zwhich)rI Zexpectedr1 �pr r r � available s
r� )N)N)N)"r� rq r4 ZloggingrM r� Ztypingr r Z cloudinitr r r r Z
cloudinit.netr r r
r r Zcloudinit.net.network_stater
r rP Z getLoggerr� r5 r r r> rE rS rT r� r r r r �<module> s(