o
    x[hŸb  ã                   @   sî   d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl	m
Z
mZ d dlmZmZmZmZ d dlmZ d dlmZ dZdZd	Zd
Ze e¡ZG dd„ dƒZG dd„ dejƒZddd„Zddd„Zdd„ Z ddd„Z!ddd„Z"dS )é    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/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 Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	dd„ Z
defdd„Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdeddfdd „Zd!ee ddfd"d#„Zd$d%„ Zd&d'„ Zed(d)„ ƒZd*d+„ Zd,d-„ ZdS ).ÚNMConnectionz/Represents a NetworkManager connection profile.c                 C   sR   t  d¡}t ¡ | _t| j_d|› �tt  ||¡ƒddœ| jd< ddi| jd< 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 Ú120)ÚidÚuuidzautoconnect-priorityÚ
connectionz%org.freedesktop.NetworkManager.originz
cloud-initÚuserN)r   ÚUUIDÚconfigparserÚConfigParserÚconfigÚstrÚoptionxformÚuuid5)ÚselfÚcon_idÚ
CI_NM_UUID© r   ú?/usr/lib/python3/dist-packages/cloudinit/net/network_manager.pyÚ__init__-   s   

ýÿzNMConnection.__init__c                 C   s:   | j  |¡si | j |< | j  ||¡s|| j | |< dS dS )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_defaultE   s
   
ÿzNMConnection._set_defaultc                 C   s   | j  |¡o| 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   Ú_config_option_is_setP   s   ÿz"NMConnection._config_option_is_setc                 C   s   |   ||¡r| j| | S dS )z]
        Returns the value of a config option if its set,
        else returns None.
        N©r(   r   r'   r   r   r   Ú_get_config_optionY   s   zNMConnection._get_config_optionc                 C   s"   |   ||¡r|| j| |< dS dS )zŒ
        Overrides the value of a config option if its already set.
        Else, if the config option is not set, it does nothing.
        Nr)   r"   r   r   r   Ú_change_set_config_optionc   s   ÿz&NMConnection._change_set_config_optionc                 C   sR   dD ]}|   |d¡dkr dS |   |d¡dvr dS qdD ]	}|  |dd¡ qdS )zt
        If for both ipv4 and ipv6, 'may-fail' is set to be False,
        set it to True for both of them.
        ©Úipv4Úipv6úmay-failÚfalseNÚmethod)ÚdhcpÚautoÚtrue)r*   r+   )r   Úfamilyr   r   r   Ú$_set_mayfail_true_if_both_false_dhcpk   s   û
ÿz1NMConnection._set_mayfail_true_if_both_false_dhcpc              	   C   sÈ   dddddddddœ}d}|   |d|¡ z|r|| }W n ty.   d}d| j| d< Y nw | j| d dkr:d	S | j| d dkrI|dkrId	S |d
v rT|   ddd¡ || j| d< |   |dd¡ d	S )z}
        Ensures there's appropriate [ipv4]/[ipv6] for given family
        appropriate for given configuration type
        Úmanualr3   r2   )ÚstaticÚstatic6Údhcp6Ú
ipv6_slaacúipv6_dhcpv6-statelessúipv6_dhcpv6-statefulÚdhcp4r2   Údisabledr1   r4   r/   N)r=   r<   r;   r-   r0   )r&   ÚKeyErrorr   )r   r5   Úsubnet_typeÚ
method_mapr1   r   r   r   Ú_set_ip_methodƒ   s6   ø€ýzNMConnection._set_ip_methodÚreturnc                 C   sL   | j  |¡si | j |< t d¡D ]}|› |› �}| j  ||¡s#|  S qdS )Né   Únot_possible)r   r    Ú	itertoolsÚcountr!   )r   r#   Ú
key_prefixÚindexÚkeyr   r   r   Ú_get_next_numbered_sectionÀ   s   
ÿz'NMConnection._get_next_numbered_sectionc                 C   s   |   ||¡}|| j| |< dS )z…
        Adds a numbered property, such as address<n> or route<n>, ensuring
        the appropriate value gets used for <n>.
        N)rL   r   )r   r#   rI   r%   rK   r   r   r   Ú_add_numberedÉ   s   zNMConnection._add_numberedc                 C   sN   |› d�}| j |  |¡}|r|› d|› d|› �n|› d|› �| j | |< dS )a:  Add route options to a given route

        Example:
        Given:
          section: ipv4
          route: route0
          key: mtu
          value: 500

        Create line under [ipv4] section:
            route0_options=mtu=500

        If the line already exists, then append the new key/value pair
        Ú_optionsú,ú=N)r   Úget)r   r#   ÚrouterK   r%   Únumbered_keyÚroute_optionsr   r   r   Ú_add_route_optionsÑ   s   
ÿýzNMConnection._add_route_optionsc                 C   s*   |d d t |d ƒ }|  |d|¡ dS )z5
        Adds an ipv[46]address<n> property.
        Úaddressú/ÚprefixN)r   rM   )r   r5   Úsubnetr%   r   r   r   Ú_add_addressè   s   zNMConnection._add_addressc                 C   s   d}|d dko|d |v S )N)z::z0.0.0.0rX   r   Únetworkr   )r   rR   Údefault_netsr   r   r   Ú_is_default_routeð   s   zNMConnection._is_default_routec                 C   s¼   t |d ƒrdnd}|d › d|d › �}d|v r"|d|d › �7 }d|v r?|  |¡r6|d › | j| d	< n	|d|d › �7 }|  |d
¡}|| j| |< d|v r\|  ||d|d ¡ dS dS )z!Adds a ipv[46].route<n> property.r[   r.   r-   rW   rX   ÚgatewayrO   Úmetriczroute-metricrR   ÚmtuN)r   r]   r   rL   rU   )r   rR   r5   r%   Ú	route_keyr   r   r   Ú
_add_routeô   s   
ÿzNMConnection._add_routeÚdnsNc                 C   sf   t |ƒrdnd}| j |¡r/|  |d¡dkr1|  |dd¡ | j| d | d | j| d< dS dS dS )	zF
        Extends the ipv[46].dns property with a name server.
        r.   r-   r1   r?   rc   Ú ú;N)r   r   r    r*   r&   )r   rc   r5   r   r   r   Ú_add_nameserver  s   
ÿ$ûzNMConnection._add_nameserverÚ
dns_searchc                 C   s^   dD ]*}| j  |¡r,|  |d¡dkr,|  |dd¡ | j | d d |¡ d | j | d< qdS )zM
        Extends the ipv[46].dns-search property with a name server.
        r,   r1   r?   z
dns-searchrd   re   N)r   r    r*   r&   Újoin)r   rg   r5   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"  ó   zNMConnection.con_uuidc                 C   s   | j  dd¡S )zN
        Can this be serialized into a meaningful connection profile?
        r   Útype)r   r!   rj   r   r   r   Úvalid(  rl   zNMConnection.validc                 C   s   |   dd¡ ¡ S )z)
        Sanitize a MAC address.
        ú-ú:)ÚreplaceÚupper)Úaddrr   r   r   Úmac_addr.  s   zNMConnection.mac_addrc                    s*  dddddddœ}||d  }|du rdS d	|v rd}nd}|| j d
 d< |dur>|| j d
 d< | ||d  ¡| j d
 d< i | j |< i dd“dd“dd“dd“dd“dd“dd“dd“dd“d d!“d"d#“d$d%“d&d'“d(d)“d*d+“d,d-“d.d/“d0d1d2d3d4d5d6d7d8d9d:d;œ¥d<d=d>œd?d@ii i dAœ}|dB }d}	g ‰g ‰ |dkr®|dC s®dDD ]}
|  |
d¡ q¥|dC D ]Y}t|ƒrºdEndF}
|  |
|d ¡ dG|v rÎ|  |
|¡ dH|v rÛ|dH | j |
 dH< |dI D ]}|  |¡ qßdJ|v ròˆ |dJ ¡ dK|v rýˆ  |dK ¡ |
dFk�rdB|v �r|dB }	q²dL|v �r/ˆ‡fdMdN„|dL dO D ƒ7 ‰ˆ ‡ fdPdN„|dL dQ D ƒ7 ‰ ˆ�s9|j�r9|j‰ˆ �sC|j�rC|j‰ ˆD ]}|  	|¡ �qEˆ �rV|  
ˆ ¡ |  ¡  |	du �ra|}	|	|k�spt dR|dS ||	¡ ||  ¡ D ]8\}}||v�r��qv|| du �rŠ�qvt|| tƒ�r¢|| �r™dTndU| j | |< �qvt|| ƒ| j | |< �qv|dk�rå|dV dWu �rÄtdXƒ| j d dY< |	du�rÒt|	ƒ| j d dB< |dZ du�rå|  |dZ ¡| j d d[< |dk�rûd\|v �rû| |d\ ¡| j d d]< |dk�r|	du�rd| j v�ri | j d< t|	ƒ| j d dB< |dk�rN|d^ D ]}| |¡}| d
dd¡ | d
d|  ¡ ¡ �q"|dZ du�rN|  |dZ ¡| j d d[< |dk�r{|	du�r{d_| j d d`< t|	ƒ| j d dB< |dZ du�r{|  |dZ ¡| j d d[< |dk�sˆ| j  |d[¡�s“|dS | j d
 da< dS dS )bzŒ
        Integrate information from network state interface information
        into the connection. Most of the work is done here.
        ÚethernetÚvlanÚbondÚbridgeÚ
infinibandN)Úphysicalrv   rw   rx   ry   Úloopbackrm   zbond-masterr   z
slave-typez-masterÚmasterÚmodez	bond-modeÚmiimonzbond-miimonÚxmit_hash_policyzbond-xmit_hash_policyÚnum_grat_arpzbond-num_grat_arpÚ	downdelayzbond-downdelayÚupdelayzbond-updelayÚfail_over_maczbond-fail_over_macÚprimary_reselectzbond-primary_reselectÚprimaryzbond-primaryÚactive_slavezbond-active_slaveÚad_actor_sys_priozbond-ad_actor_sys_prioÚad_actor_systemzbond-ad_actor_systemÚ	ad_selectzbond-ad_selectÚad_user_port_keyzbond-ad_user_port_keyÚall_slaves_activezbond-all_slaves_activeÚarp_all_targetszbond-arp_all_targetsÚarp_intervalzbond-arp_intervalzbond-arp_ip_targetzbond-arp_validatezbond-lacp_ratezbond-lp_intervalzbond-min_linkszbond-num_unsol_nazbond-packets_per_slavezbond-peer_notif_delayzbond-resend_igmpzbond-tlb_dynamic_lbzbond-use_carrier)Úarp_ip_targetÚarp_validateÚ	lacp_rateÚlp_intervalÚ	min_linksÚnum_unsol_naÚpackets_per_slaveÚpeer_notif_delayÚresend_igmpÚtlb_dynamic_lbÚuse_carrierÚ
bridge_stpÚbridge_bridgeprio)ÚstpÚpriorityr   Úvlan_id)rw   rx   rv   ru   ry   r`   Úsubnetsr,   r.   r-   rV   r^   ÚroutesÚdns_nameserversrg   rc   c                    ó   g | ]}|ˆ vr|‘qS r   r   )Ú.0rc   )Úfound_nameserversr   r   Ú
<listcomp>¿  ó
    þz1NMConnection.render_interface.<locals>.<listcomp>Únameserversc                    r¡   r   r   )r¢   Úsearch)Úfound_dns_searchr   r   r¤   Ä  r¥   r§   zZNetwork config: ignoring %s device-level mtu:%s because ipv4 subnet-level mtu:%s provided.Únamer4   r0   Ú	wakeonlanTé@   zwake-on-lanÚmac_addresszmac-addresszvlan-raw-deviceÚparentÚbridge_portsÚdatagramztransport-modezinterface-name)r   Úcon_refrC   r	   rZ   rb   Úextendr    Údns_searchdomainsrf   ri   r6   ÚLOGÚwarningÚitemsÚ
isinstanceÚboolr   rt   Úget_connr&   rk   r!   )r   ÚifaceÚnetwork_stater   Ú	_type_mapÚif_typeÚ
slave_typeÚ	_prop_mapÚ
device_mtuÚipv4_mtur5   rY   rR   Ú
nameserverÚnm_proprK   Úportr   )r¨   r£   r   Úrender_interface5  sT  ú	
ÿ
ÿþüúùø	÷õóñïíëéèçæÓ0þÿÈ;€

þ
þ


û	
ÿ

ÿÿ


ÿÿÿýzNMConnection.render_interfacec                 C   s(   t  ¡ }| jj|dd� d}|| ¡  S )z
        Stringify.
        F)Úspace_around_delimitersz2# Generated by cloud-init. Changes will be lost.

)ÚioÚStringIOr   ÚwriteÚgetvalue)r   ÚbufÚheaderr   r   r   Údump"  s   zNMConnection.dump)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r&   r(   r*   r+   r6   rC   r   rL   rM   rU   rZ   r]   rb   rf   r   ri   rk   rn   Ústaticmethodrt   rÄ   rÌ   r   r   r   r   r   *   s0    	
=	
 nr   c                   @   sJ   e Zd ZdZddd„Zdd„ Zdd„ Z		dd	ed
ee	 ddfdd„Z
dS )ÚRendererz‘Renders network information in a NetworkManager keyfile format.

    See https://networkmanager.dev/docs/api/latest/nm-settings-keyfile.html
    Nc                 C   s   i | _ || _d S ©N)Úconnectionsr   )r   r   r   r   r   r   3  s   
zRenderer.__init__c                 C   s
   | j | S rÓ   )rÔ   ©r   r   r   r   r   r¸   7  s   
zRenderer.get_connc                 C   s   || j v r| j |  ¡ S |S rÓ   )rÔ   rk   rÕ   r   r   r   r°   :  s   
zRenderer.con_refrº   Ú	templatesrD   c           	      C   sº   |  ¡ D ]}| d¡p|d }t|d ƒ| j|< q|  ¡ D ]}| d¡p'|d }| j| }| ||| ¡ q| j ¡ D ]\}}| ¡ sCq:t||ƒ}t 	|| 
¡ d¡ q:t 	t|ƒtd¡ d S )NÚ	config_idr©   i€  )Úiter_interfacesrQ   r   rÔ   rÄ   rµ   rn   Únm_conn_filenamer   Ú
write_filerÌ   Úcloud_init_nm_conf_filenameÚNM_IPV6_ADDR_GEN_CONF)	r   rº   rÖ   Útargetr¹   Úconn_keyÚconnr   r©   r   r   r   Úrender_network_stateA  s   	


ÿzRenderer.render_network_staterÓ   )NN)rÍ   rÎ   rÏ   rÐ   r   r¸   r°   r
   r   Údictrà   r   r   r   r   rÒ   -  s    

üþýûrÒ   c                 C   s&   t  |t¡}d| › d�}|› d|› �S )Nzcloud-init-z.nmconnectionz/system-connections/©r   Útarget_pathÚ
NM_RUN_DIR)r   rÝ   Útarget_con_dirÚcon_filer   r   r   rÙ   b  s   rÙ   c                 C   s$   t  |t¡}d| › �}|› d|› �S )Nzifcfg-rW   )r   rã   ÚIFCFG_CFG_FILE)ÚdevnamerÝ   rå   ræ   r   r   r   Úsysconfig_conn_filenameh  s   
ré   c                 C   s6   t | ƒ}tj |¡stƒ rt| ƒ}tj |¡r|S dS )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)rÙ   ÚosÚpathÚisfiler   ré   )rè   Ú	conn_filer   r   r   Úconn_filenamen  s   rî   c                 C   s   t  | t¡}d}|› d|› �S )Nz$30-cloud-init-ip6-addr-gen-mode.confz/conf.d/râ   )rÝ   rå   Ú	conf_filer   r   r   rÛ   ƒ  s   rÛ   c                 C   s\   ddl m} tjd| d�}d}|ƒ r(z	t g d¢¡ W n tjy'   d}Y nw t|ƒo-|S )Nr   )Úuses_systemdÚnmcli)rÝ   T)Ú	systemctlz
is-enabledzNetworkManager.serviceF)Úcloudinit.distrosrð   r   ÚwhichÚProcessExecutionErrorr·   )rÝ   rð   Únmcli_presentÚservice_activer   r   r   Ú	available‰  s   ÿrø   rÓ   )#r   rÆ   rG   Úloggingrê   r   Útypingr   r   Ú	cloudinitr   r   Úcloudinit.netr   r   r   r	   Úcloudinit.net.network_stater
   Úcloudinit.net.sysconfigr   rä   Ú
NM_LIB_DIRrç   rÜ   Ú	getLoggerrÍ   r³   r   rÒ   rÙ   ré   rî   rÛ   rø   r   r   r   r   Ú<module>   s6   
	    
5

