3
\�me�0 � @ s� d dl Z d dlmZ d dlmZ d dlmZmZ d dlm Z d dl
m Z e j e
�ZG dd� d�ZG d d
� d
e j�Zd
d d �ZdS )� N)� OrderedDict)�Optional)�subp�util)�renderer)� NetworkStatec @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� CfgParserc C s t g g g g g g i d��| _d S )N)�Match�Link�Network�DHCPv4�DHCPv6�Address�Route)r � conf_dict)�self� r � /usr/lib/python3.6/networkd.py�__init__ s zCfgParser.__init__c C sd x^| j j� D ]P}||kr | j | j|d t|� � ttj| j | ��| j |< | j | j� q W d S )N�=)r �keys�append�str�list�dict�fromkeys�sort)r �sec�key�val�kr r r �update_section s
zCfgParser.update_sectionc C s� x�| j j� D ]|}||kr || j | kr4g | j | |< | j | | j|d t|� � ttj| j | | ��| j | |< | j | | j� q W dS )z�
For each route section we use rid as a key, this allows us to isolate
this route from others on subsequent calls.
r N)r r r r r r r r )r r �ridr r r r r r �update_route_section( s z CfgParser.update_route_sectionc C s� d}x�t | jj� �D ]�\}}|s"q|dkrbx�t |�D ](}|d| d 7 }||d 7 }|d7 }q4W q|dkr�x�t |�D ]>}|d| d 7 }x(t || �D ]}||d 7 }|d7 }q�W qtW q|d| d 7 }x t |�D ]}||d 7 }q�W |d7 }qW |S )N� r �[z]
�
r )�sortedr �items)r �contentsr �v�e�nr r r �get_final_conf8 s( zCfgParser.get_final_confN)�__name__�
__module__� __qualname__r r! r# r- r r r r r s
r c @ s� e Zd ZdZd dd�Zed�dd�Zed�dd �Zed�d
d �Zed�d d
�Z ed�dd�Z
ee d�dd�Z ed�dd�Z
dd� Zd e ee dd�dd�Ze ed�dd �ZdS ) �Rendererz�
Renders network information in /etc/systemd/network
This Renderer is currently experimental and doesn't support all the
use cases supported by the other renderers yet.
Nc C s( |si }|j dd�| _|j dd�| _d S )N�resolve_conf_fnz/etc/systemd/resolved.conf�network_conf_dirz/etc/systemd/network/)�getr2 r3 )r �configr r r r Y s
zRenderer.__init__)�cfgc C sX d}dddd�}|sd S x6|j � D ]*\}}||kr"|| r"|j|||| � q"W |d S )Nr �NameZDriverZ
MACAddress)�nameZdriverZ mac_addressr8 )r( r! )r �ifacer6 r Z
match_dictr r* r r r �generate_match_sectionc s z Renderer.generate_match_sectionc C s2 d}|s d S d|kr.|d r.|j |d|d � d S )Nr
ZmtuZMTUBytes)r! )r r9 r6 r r r r �generate_link_sectiont s
z Renderer.generate_link_sectionc C st d}dddd�}d}d|kr,dt |d � }xB|j� D ]6\}}||krHq6|d krX||7 }|j|||| |� q6W d
S ) zy
Parse a route and use rid as a key in order to isolate the route from
others in the route dict.
r �GatewayZ DestinationZMetric)�gateway�networkZmetricr$ �prefix�/r> N)r r( r# ) r r" �confr6 r Z
route_cfg_mapr? r r* r r r � parse_routes} s zRenderer.parse_routesc C s� d}d}d}�xj|j dg �D �]X}|d }|dks:|dkrV|dkrHd}qx|d krxd
}n"|d krx|dkrld }n |dkrxd
}d |kr�|d r�x,|d D ] }| jd
|� �||� |d }q�W d|kr ddddd�} x�|j� D ]�\}
} |
dk�rd|k�r| dt|d � 7 } |jd| |
| � q�|
dk�rH|jdd|� �| |
| � |d }q�|
dk�sZ|
dkr�|j|| |
d j| �� q�W q W |j|d |� |d!k�r�t|j d d �t��r�|j|d |d � |S )"N�nor r Zsubnets�typeZdhcp4�dhcpZipv4�ipv6�yesZdhcp6Zroutes�r� �addressr r< �DNS�Domains)rJ r= �dns_nameservers�
dns_searchr? r@ r= r �arM rN � ZDHCPz accept-rar$ Z IPv6AcceptRA)rF rG ) r4 rB r( r r! r# �join�
isinstance�bool) r r9 r6 rE r r" r+ �t�iZsubnet_cfg_mapr r* r r r �
parse_subnets� sP
zRenderer.parse_subnetsc
C s� x�d
D ]�}||krqt || �j� }ytj|�}|r8dnd}W n, tk
rj |dkrftjd|� d}Y nX |dkrzd}nd}|j|d |� qW d S ) N� dhcp4domain� dhcp6domainrG rC �routez Invalid dhcp4domain value - %sr r
�
UseDomains)rW rX )r �casefoldr Ztranslate_bool�
ValueError�LOGZwarningr! )r �dr6 �item�retZsectionr r r � dhcp_domain� s
zRenderer.dhcp_domain)r6 �nsc C s� d}dddd�}|j d�}| r:|jdkr:|j|jd�}n| rN|jdkrNd S x<|j� D ]0\}}||krX|| rX|j||d j|| �� qXW d S )
Nr rL rK )�search� nameserversZ addresses�dnsrI )rc rd � rP )r4 �versionZdns_searchdomainsrM r( r! rQ ) r r9 r6 rb r Z dns_cfg_mapre r r* r r r � parse_dns� s
zRenderer.parse_dnsc C s� ddddd�}|dkr,|j ddd d
d d �� d
|� d�|kr�|dd|� �gkr�|d
|� d� }x4|j� D ](\}}||krh|jd|� �||| � qhW d S )Nzuse-dnsz use-domainsz use-hostnamezuse-ntp)ZUseDNSrZ Z UseHostnameZUseNTP�4z
send-hostnameZhostnamez route-metriczuse-mtuz
use-routes)Z SendHostnameZHostnameZ RouteMetricZUseMTUZ UseRoutesrE z
-overridesrG ZipvZDHCPv)�updater( r! ) r r6 �devicerE rg Zdhcp_config_mapsZdhcp_overridesr r* r r r �parse_dhcp_overrides� s$
z Renderer.parse_dhcp_overridesc C s> d}t jd|� |d | d }tj||� tj|||� d S )Nzsystemd-networkz Setting Networking Config for %sz10-cloud-init-z.network)r] �debugr Z
write_fileZ chownbyname)r �linkrA Znwk_dirZ net_fn_ownerZnet_fnr r r �create_network_file s
z Renderer.create_network_file)�
network_state� templates�returnc C sT | j }|rtj|�| }tj|� | j|�}x"|j� D ]\}}| j|||� q6W d S )N)r3 r Z target_pathr Z
ensure_dir�_render_contentr( ro )r rp rq �targetZ network_dir�ret_dictr r* r r r �render_network_state s
z Renderer.render_network_state)rb rr c C sz i }�xn|j � D �]`}t� }| j||�}| j||� | j||�}| j|||� d}x,|j� D ] }| jd|� �||� |d }q\W |jdk�r`|d } x6|j d j
� D ]$\}
} d| kr�| j d�| kr�|
} P q�W | |j d k�r`|j d | } | j | |� xhdD ]`}
d
|
� d �| k�rJd | j d
|
� d
�i �k�rJ| � d|
� d|
� d�}t
|��| j|| ||
� � q�W |j||j� i� qW |S )Nr �crI rf r8 Z ethernetszset-nameri �6rE Zdomainz use-domainsz
-overridesz has both dhcpzdomain and dhcpz*-overrides.use-domains configured. Use one)ri rx )Ziter_interfacesr r: r; rV rh Z iter_routesrB rg r5 r( r4 ra � RuntimeErrorrl rj r- )r rb ru r9 r6 rn rE r" rY r8 Zdev_nameZdev_cfgrk rg Z exceptionr r r rs ( s<
zRenderer._render_content)N)NN)r. r/ r0 �__doc__r r r: r; rB rV ra r rh rl ro r r rv rs r r r r r1 Q s
6
r1 c C s6 ddg}ddg}x |D ]}t j||| d�sdS qW dS )NZipZ systemctlz /usr/sbinz/bin)rc rt FT)r Zwhich)rt Zexpectedrc �pr r r � availablec s
r| )N)Zlogging� collectionsr Ztypingr Z cloudinitr r Z
cloudinit.netr Zcloudinit.net.network_stater Z getLoggerr. r] r r1 r| r r r r �<module> s
?