CasperSecurity
U
P�b_0 � @ s` d Z ddlZddlZddlZddlZddlZddlZddlZG dd� de�Z G dd� de
�Z dS )z netplan configuration manager� Nc @ sJ e Zd Zdi fdd�Zedd� �Zedd� �Zedd � �Zed
d � �Zed d
� �Z edd� �Z
edd� �Z edd� �Z edd� �Z
edd� �Zedd� �Zedd� �Zed d � �Zed d � �Zed d!� �Zed"d#� �Zed$d%� �Zed&d'� �Zg fd(d)�Zd*d+� Zd?d-d.�Zd/d0� Zd1d2� Z d3d4� Z d@d6d7�Z d8d9� Z d:d;� Z d<d=� Z!d>S )A�
ConfigManager�/c C sP || _ tjdd�| _tj�| jd�| _tj�| jd�| _|| _ i | _
t � | _ d S )NZnetplan_)�prefix�etc�run)
r �tempfile�mkdtemp�tempdir�os�path�join�temp_etc�temp_run� extra_files�config�set�new_interfaces)�selfr r � r �+/usr/share/netplan/netplan/configmanager.py�__init__ s zConfigManager.__init__c C s
| j d S )N�network)r �r r r r r ( s zConfigManager.networkc C sh i }|� | j� |� | j� |� | j� |� | j� |� | j� |� | j� |� | j� |� | j� |S �N) �update� ovs_ports� ethernets�modems�wifis�bridges�bonds�tunnels�vlans�r �
interfacesr r r r$ , s zConfigManager.interfacesc C s, i }|� | j� |� | j� |� | j� |S r )r r r r r# r r r �physical_interfaces9 s
z!ConfigManager.physical_interfacesc C s8 i }|� | j� |� | j� |� | j� |� | j� |S r )r r r r! r" r# r r r �virtual_interfacesA s z ConfigManager.virtual_interfacesc C s
| j d S )Nr �r r r r r r K s zConfigManager.ovs_portsc C s
| j d S )N� openvswitchr' r r r r r( O s zConfigManager.openvswitchc C s
| j d S )Nr r' r r r r r S s zConfigManager.ethernetsc C s
| j d S )Nr r' r r r r r W s zConfigManager.modemsc C s
| j d S )Nr r' r r r r r [ s zConfigManager.wifisc C s
| j d S )Nr r' r r r r r _ s zConfigManager.bridgesc C s
| j d S )Nr r' r r r r r c s zConfigManager.bondsc C s
| j d S )Nr! r' r r r r r! g s zConfigManager.tunnelsc C s
| j d S )Nr" r' r r r r r" k s zConfigManager.vlansc C s
| j d S )N�
nm-devicesr' r r r r �
nm_deviceso s zConfigManager.nm_devicesc C s
| j d S )N�versionr' r r r r r+ s s zConfigManager.versionc C s
| j d S )N�rendererr' r r r r r, w s zConfigManager.rendererc C s | � | j�S r )�
strip_treer r r r r �tree{ s zConfigManager.treec C sD i }| � � D ]2\}}t|t�r(t�|�}|ddi fkr |||< q |S )zclear empty branches� N)�items�
isinstance�dictr r- )�dataZnew_data�k�vr r r r- s
zConfigManager.strip_treec
s� i � dD ]4}t � tj�| j|dd��D ]}|� tj�|�< q&q� fdd�t� �� �D �}i i i i i i i i i i ddd� | jd< |D ]}| � |� q�|D ]}| j
| � |�O _
q�t � d �
tj| jd
d ��� dS ) z�
Parse all our config files to return an object that describes the system's
entire configuration, so that it can later be interrogated.
Returns a dict that contains the entire, collated and merged YAML.
)�libr r �netplanz*.yamlc s g | ] }� | �qS r r )�.0�name�Znames_to_pathsr r �
<listcomp>� s z'ConfigManager.parse.<locals>.<listcomp>N) r r( r r r r r r! r" r) r+ r, r zMerged config:
{}F)Zdefault_flow_style)�globr
r r r �basename�sorted�keysr �_merge_yaml_configr �logging�debug�format�yaml�dumpr. )r Z extra_configZyaml_dir� yaml_file�filesr r: r �parse� s. � zConfigManager.parsec C s* |D ]}| � ||| � q| j�|� d S r )�
_copy_filer r )r Z config_dictZ config_filer r r �add� s zConfigManager.addTc C s� |r(| � tj�| jd�tj�| jd�� | j tj�| jd�tj�| jdd�dd� | j tj�| jd�tj�| jd d
�dd� d S ) Nz etc/netplanr7 �%run/NetworkManager/system-connectionsZNetworkManagerzsystem-connectionsT)�
missing_ok�run/systemd/networkZsystemdr )�
_copy_treer
r r r r
r )r Zbackup_config_dirr r r �backup� s ���zConfigManager.backupc
C s z�t | j�D ] }t�| j| � | j|= q d�| j�}d�| j�}tj�|�r|t� tj�
| j d�� | � |tj�
| j d�� tj�|�r�t� tj�
| j d�� | � |tj�
| j d�� W nF t
k
r� } z(t�d�|�� t�d� t�d� W 5 d }~X Y nX d S )Nz${}/NetworkManager/system-connectionsz{}/systemd/networkrK rM z8Something really bad happened while reverting config: {}zZYou should verify the netplan YAML in /etc/netplan and probably run 'netplan apply' again.���)r2 r r
�unlinkrC r r �exists�shutil�rmtreer r rN � ExceptionrA �error�sys�exit)r Z
extra_fileZ temp_nm_pathZtemp_networkd_path�er r r �revert� s(
� �
zConfigManager.revertc C s t �| j� d S r )rS rT r r r r r �cleanup� s zConfigManager.cleanupc C s t �||� d S r )rS �copy)r �src�dstr r r rI � s zConfigManager._copy_fileFc C s2 zt �||� W n tk
r, |r&n� Y nX d S r )rS �copytree�FileNotFoundError)r r] r^ rL r r r rN � s zConfigManager._copy_treec
C s� t � }t� }d|krD|�d�D ]$\}}|i d�||< |i d�||< q t|�� �}|D ]\}|�|�} ||kr�t�d�||�� || � | � qTt�d�||�� | ||< |�
|� qT|S )N�ports)Zpeerr( �{} exists in {}�{} not found in {}) r r2 �get�listr? �poprA rB rC r rJ )
r �orig�newr ra Zp1Zp2�changed_ifaces�ifname�ifacer r r �_merge_ovs_ports_config� s
z%ConfigManager._merge_ovs_ports_configc C sx t � }t|�� �}|D ]\}|�|�}||krNt�d�||�� || �|� qt�d�||�� |||< |�|� q|S )Nrb rc ) r re r? rf rA rB rC r rJ )r rg rh r ri rj rk r r r �_merge_interface_config s
z%ConfigManager._merge_interface_configc C s t � }�z�t|����}tj|tjd�}d }|d k r<|�d�}|�r�d|krv| �| j|�d��}||O }|�d�| jd< d|kr�| � | j
|�d��}||O }d|kr�| � | j |�d��}||O }d|kr�| � | j |�d��}||O }d|k�r| � | j
|�d��}||O }d|k�r.| � | j|�d��}||O }d |k�rT| � | j|�d ��}||O }d
|k�rz| � | j|�d
��}||O }d |k�r�| � | j|�d ��}||O }d |k�r�|�d �| jd < d
|k�r�|�d
�| jd
< W 5 Q R X |W S ttjfk
�r t�d�|�� t�d� Y nX d S )N)�Loaderr r( r r r r r r! r" r) r+ r, z!Error while loading {}, aborting.� )r �openrD �loadZ CSafeLoaderrd rl r r rm r r r r r r! r" r* �IOErrorZ YAMLErrorrA rV rC rW rX )r rF r �fZ yaml_datar rh r r r r@ sX
z ConfigManager._merge_yaml_configN)T)F)"�__name__�
__module__� __qualname__r �propertyr r$ r% r& r r( r r r r r r! r" r* r+ r, r. � staticmethodr- rH rJ rO rZ r[ rI rN rl rm r@ r r r r r s^
,
r c @ s e Zd ZdZdS )�ConfigurationErrorzL
Configuration could not be parsed or has otherwise failed to apply
N)rt ru rv �__doc__r r r r ry F s ry ) rz r< rA r
rS rW r rD �objectr rU ry r r r r �<module> s +