3
�~�f9L � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
m Z d dl m
Z
mZmZmZ d dlmZ d dlmZ dZdZd Zd
Zd Zeje�Z G d d
� d
�Z G dd� dej �Z ddd�Z ddd�Z dd� Z!d dd�Z"d dd�Z#dS ) � N)�List�Optional)�subp�util)�is_ipv6_address�is_ipv6_network�renderer�subnet_is_ipv6)� NetworkState)�available_nm_ifcfg_rhz/etc/NetworkManagerz/usr/lib/NetworkManagerz'/etc/NetworkManager/NetworkManager.confz /etc/sysconfig/network-scriptsz�# This is generated by cloud-init. Do not edit.
#
[.config]
enable=nm-version-min:1.40
[connection.30-cloud-init-ip6-addr-gen-mode]
# Select EUI64 to be used if the profile does not specify it.
ipv6.addr-gen-mode=0
c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
edd�dd�Zee dd�dd �Zd d � Zd d � Zed!d"� �Zd#d$� Zd%d&� ZdS )'� NMConnectionz/Represents a NetworkManager connection profile.c C sR t jd�}tj� | _t| j_d|� �tt j||��dd�| jd< ddi| jd< d S )
z�
Initializes the connection with some very basic properties,
notably the UUID so that the connection can be referred to.
z$a3924cb8-09e0-43e9-890b-77972a800108z cloud-init Z120)�id�uuidzautoconnect-priority�
connectionz%org.freedesktop.NetworkManager.originz
cloud-init�userN)r ZUUID� configparserZ ConfigParser�config�strZ optionxformZuuid5)�self�con_idZ
CI_NM_UUID� r �%/usr/lib/python3.6/network_manager.py�__init__. s
zNMConnection.__init__c C s6 | j j|�si | j |< | j j||�s2|| j | |< dS )z_
Sets a property unless it's already set, ensuring the section
exists.
N)r � has_section�
has_option)r �section�option�valuer r r � _set_defaultF s
zNMConnection._set_defaultc C s | j j|�o| j j||�S )zf
Checks if a config option is set. Returns True if it is,
else returns False.
)r r r )r r r r r r �_config_option_is_setQ s z"NMConnection._config_option_is_setc C s" | j ||�r| j| | S dS dS )z]
Returns the value of a config option if its set,
else returns None.
N)r r )r r r r r r �_get_config_optionZ s z NMConnection._get_config_optionc C s | j ||�r|| j| |< dS )z�
Overrides the value of a config option if its already set.
Else, if the config option is not set, it does nothing.
N)r r )r r r r r r r �_change_set_config_optiond s z&NMConnection._change_set_config_optionc C sV x4d
D ],}| j |d�dkr dS | j |d�d krdS qW xd D ]}| j|dd � q<W dS )
zt
If for both ipv4 and ipv6, 'may-fail' is set to be False,
set it to True for both of them.
�ipv4�ipv6zmay-fail�falseN�method�dhcp�auto�true)r" r# )r& r' )r" r# )r r! )r �familyr r r �$_set_mayfail_true_if_both_false_dhcpl s
z1NMConnection._set_mayfail_true_if_both_false_dhcpc C s� ddddddddd�}| j |dd� y || }W n& tk
rV d}d| j| d< Y nX | j| d dkrnd S | j| d dkr�|dkr�d S |dkr�| j d
dd� || j| d< | j |dd� d S )z}
Ensures there's appropriate [ipv4]/[ipv6] for given family
appropriate for given configuration type
Zmanualr' r& )ZstaticZstatic6Zdhcp6�
ipv6_slaaczipv6_dhcpv6-statelesszipv6_dhcpv6-statefulZdhcp4r& r% Zdisabledr( zmay-failN�ipv6_dhcpv6-stateful�ipv6_dhcpv6-statelessr+ r" r$ )r, r- r+ )r �KeyErrorr )r r) Z subnet_typeZ
method_mapr% r r r �_set_ip_method� s2 zNMConnection._set_ip_methodc C sX | j j|�si | j |< x<tjd�D ].}|� |� �}| j j||�s"|| j | |< P q"W dS )z�
Adds a numbered property, such as address<n> or route<n>, ensuring
the appropriate value gets used for <n>.
� N)r r � itertools�countr )r r Z
key_prefixr �index�keyr r r �
_add_numbered� s
zNMConnection._add_numberedc C s* |d d t |d � }| j|d|� dS )z5
Adds an ipv[46]address<n> property.
�address�/�prefixN)r r5 )r r) �subnetr r r r � _add_address� s zNMConnection._add_addressc C sV t |d �rdnd}|d d t|d � }d|krD|d |d }| j|d|� d S )
z!Adds a ipv[46].route<n> property.�gatewayr# r" Znetworkr7 r8 �,�routeN)r r r5 )r r= r) r r r r �
_add_route� s
zNMConnection._add_routeN)�dns�returnc C sN t |�r dnd}| jj|�rJ| j|dd� | j| d | d | j| d< dS )zF
Extends the ipv[46].dns property with a name server.
r# r" r? � �;N)r r r r )r r? r) r r r �_add_nameserver� s z NMConnection._add_nameserver)�
dns_searchr@ c C sR xLdD ]D}| j j|�r| j|dd� | j | d dj|� d | j | d< qW dS )zM
Extends the ipv[46].dns-search property with a name server.
r" r# z
dns-searchrA rB N)r" r# )r r r �join)r rD r) r r r �_add_dns_search� s
z NMConnection._add_dns_searchc C s | j d d S )z-
Returns the connection UUID
r r )r )r r r r �con_uuid� s zNMConnection.con_uuidc C s | j jdd�S )zN
Can this be serialized into a meaningful connection profile?
r �type)r r )r r r r �valid� s zNMConnection.validc C s | j dd�j� S )z)
Sanitize a MAC address.
�-�:)�replace�upper)Zaddrr r r �mac_addr� s zNMConnection.mac_addrc
st ddddddd�}||d }|dkr*dS d |kr8d}nd}|| j d
d< |dk r||| j d
d < |j||d �| j d
d
< i | j |< dddddddddd� ddd�dd ii i d �}|d }d} g �g � x�|d D ]�}
t|
�r�d nd!} | j| |
d � d"|
k�r | j| |
� d#|
k�r(|
d# | j | d#< x |
d$ D ]} | j| � �q2W d%|
k�r^�j|
d% � d&|
k�rv� j|
d& � | d!kr�d |
kr�|
d } q�W d'|k�r؈�fd(d)�|d' d* D �7 �� � fd+d)�|d' d, D �7 � � �r�|j�r�|j�� �r|j�r|j� x�D ]}
| j |
� �q
W � �r.| j
� � | j � | dk�rD|} | |k�sbt j
d-|d. || � x~|| j� D ]n\}}||k�r��qp|| dk�r��qpt|| t��r�|| �r�d/nd0| j | |< nt|| �| j | |< �qpW |dk�rN|d1 d2k�r td3�| j d d4< | dk �r(t| �| j d d < |d5 dk �rN| j|d5 �| j d d6< |dk�rzd7|k�rz|j|d7 �| j d d8< |dk�r�x<|d9 D ]0}|j|�}|jd
d d� |jd
d
| j� � �q�W |d5 dk �r�| j|d5 �| j d d6< |dk�rB| dk �rBd:| j d d;< t| �| j d d < |d5 dk �rB| j|d5 �| j d d6< |dk�s^| j j|d6� �rp|d. | j d
d<< dS )=z�
Integrate information from network state interface information
into the connection. Most of the work is done here.
�ethernet�vlan�bond�bridge�
infinibandN)ZphysicalrP rQ rR rS ZloopbackrH z bond-masterr z
slave-typez-masterZmasterz bond-modeZ bond_miimonzbond-xmit-hash-policyzbond-num-grat-arpzbond-downdelayz bond-updelayzbond-fail-over-maczbond-primary-reselectz bond-primary) �modeZmiimonZxmit_hash_policyZ num_grat_arpZ downdelayZupdelayZ
fail_over_macZprimary_reselectZprimaryZ
bridge_stpZbridge_bridgeprio)ZstpZpriorityr
Zvlan_id)rQ rR rP rO rS ZmtuZsubnetsr# r" r6 r; Zroutes�dns_nameserversrD r? c s g | ]}|� kr|�qS r r )�.0r? )�found_nameserversr r �
<listcomp>O s z1NMConnection.render_interface.<locals>.<listcomp>Z nameserversc s g | ]}|� kr|�qS r r )rV �search)�found_dns_searchr r rX T s rY zZNetwork config: ignoring %s device-level mtu:%s because ipv4 subnet-level mtu:%s provided.�namer( r$ Z wakeonlanT�@ z wake-on-lanZ mac_addressz mac-addresszvlan-raw-device�parentZ bridge_portsZdatagramztransport-modezinterface-name)r �con_refr r/ r: r> �extendrU Zdns_searchdomainsrC rF r* �LOGZwarning�items�
isinstance�boolr rN �get_connr rG r )r �iface�
network_stater Z _type_mapZif_typeZ
slave_typeZ _prop_mapZ
device_mtuZipv4_mtur9 r) r= Z
nameserverZnm_propr4 Zportr )rZ rW r �render_interface� s�
z NMConnection.render_interfacec C s( t j� }| jj|dd� d}||j� S )z
Stringify.
F)Zspace_around_delimitersz2# Generated by cloud-init. Changes will be lost.
)�io�StringIOr �write�getvalue)r Zbuf�headerr r r �dump� s zNMConnection.dump)�__name__�
__module__� __qualname__�__doc__r r r r r! r* r/ r5 r: r> r rC r rF rG rI � staticmethodrN rg rm r r r r r + s&
-
5r c @ sB e Zd ZdZd dd�Zdd� Zdd� Zd
eee dd �d
d �Z
dS )�Rendererz?Renders network information in a NetworkManager keyfile format.Nc C s i | _ || _d S )N)� connectionsr )r r r r r r � s zRenderer.__init__c C s
| j | S )N)rt )r r r r r rd � s zRenderer.get_connc C s || j kr| j | j� S |S d S )N)rt rG )r r r r r r^ � s
zRenderer.con_ref)rf � templatesr@ c C s� x&|j � D ]}t|d �| j|d < q
W x,|j � D ] }| j|d }|j||| � q2W x<| jj� D ].\}}|j� stqbt||�}tj||j � d� qbW tjt
|�t d� d S )Nr[ i� ) Ziter_interfacesr rt rg ra rI �nm_conn_filenamer Z
write_filerm �cloud_init_nm_conf_filename�NM_IPV6_ADDR_GEN_CONF)r rf ru �targetre Zconnr r[ r r r �render_network_state� s
z Renderer.render_network_state)N)NN) rn ro rp rq r rd r^ r
r �dictrz r r r r rs � s
rs c C s& t j|t�}d| � d�}|� d|� �S )Nz cloud-init-z
.nmconnectionz/system-connections/)r � target_path�
NM_RUN_DIR)r ry �target_con_dir�con_filer r r rv � s rv c C s$ t j|t�}d| � �}|� d|� �S )Nzifcfg-r7 )r r| �IFCFG_CFG_FILE)�devnamery r~ r r r r �sysconfig_conn_filename� s
r� c C s8 t | �}tjj|� r$t� r$t| �}tjj|�r4|S dS )a�
This function returns the name of the interface config file.
It first checks for presence of network manager connection file.
If absent and ifcfg-rh plugin for network manager is available,
it returns the name of the ifcfg file if it is present. If the
plugin is not present or the plugin is present but ifcfg file is
not, it returns None.
This function is called from NetworkManagerActivator class in
activators.py.
N)rv �os�path�isfiler r� )r� Z conn_filer r r �
conn_filename� s r� c C s t j| t�}d}|� d|� �S )Nz$30-cloud-init-ip6-addr-gen-mode.confz/conf.d/)r r| r} )ry r~ Z conf_filer r r rw s rw c C sz ddl m} tjjtj| td��}tjd| d�}d}|� rjytjddd g� W n tj k
rh d
}Y nX |oxt
|�ox|S ) Nr )� uses_systemd)r� Znmcli)ry TZ systemctlz
is-enabledzNetworkManager.serviceF) Zcloudinit.distrosr� r� r� r� r r| � NM_CFG_FILEZwhichZProcessExecutionErrorrc )ry r� Zconfig_presentZ
nmcli_presentZservice_activer r r � available s
r� )N)N)N)N)$r rh r1 Zloggingr� r Ztypingr r Z cloudinitr r Z
cloudinit.netr r r r Zcloudinit.net.network_stater
Zcloudinit.net.sysconfigr r} Z
NM_LIB_DIRr� r� rx Z getLoggerrn r` r rs rv r� r� rw r� r r r r �<module> s4
0