3
\�meS9 � @ s� d Z ddlZddlZddlmZ ddlmZmZmZm Z m
Z
ddl j Z ddl
mZmZmZ ddlmZ eje�ZG dd� d�ZG d d
� d
�ZG d d � d �ZG d
d� d�ZdS )z.Module for ephemeral network context managers
� N)�partial)�Any�Callable�Dict�List�Optional)� IscDhclient�NoDHCPLeaseError� maybe_perform_dhcp_discovery)�ProcessExecutionErrorc @ sT e Zd ZdZdeeeef d�dd�Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dS )�EphemeralIPv4Networka� Context manager which sets up temporary static network configuration.
No operations are performed if the provided interface already has the
specified configuration.
This can be verified with the connectivity_url_data.
If unconnected, bring up the interface with valid ip, prefix and broadcast.
If router is provided setup a default route for that interface. Upon
context exit, clean up the interface leaving no configuration behind.
N)�connectivity_url_datac
C s� t ||||g�s$tdj||||���ytj|�| _W n2 tk
rf } ztdj| ��| �W Y dd} ~ X nX || _|| _|| _|| _ || _
|| _ g | _ || _
dS )aX Setup context manager and validate call signature.
@param interface: Name of the network interface to bring up.
@param ip: IP address to assign to the interface.
@param prefix_or_mask: Either netmask of the format X.X.X.X or an int
prefix.
@param broadcast: Broadcast address for the IPv4 network.
@param router: Optionally the default gateway IP.
@param connectivity_url_data: Optionally, a URL to verify if a usable
connection already exists.
@param static_routes: Optionally a list of static routes from DHCP
z5Cannot init network on {0} with {1}/{2} and bcast {3}z4Cannot setup network, invalid prefix or netmask: {0}N)�all�
ValueError�format�netZipv4_mask_to_net_prefix�prefixr
� interface�ip� broadcast�router�
static_routes� cleanup_cmds�distro)
�selfr r r �prefix_or_maskr r r
r �e� r � /usr/lib/python3.6/ephemeral.py�__init__ s& z EphemeralIPv4Network.__init__c C s| | j r(tj| j �r(tjd| j d � dS y*| j� | jrB| j� n| jrP| j � W n$ t
k
rv | j ddd� � Y nX dS )z>Perform ephemeral network setup if interface is not connected.z=Skip ephemeral network setup, instance has connectivity to %sZurlN) r
r �has_url_connectivity�LOG�debug�_bringup_devicer �_bringup_static_routesr �_bringup_routerr �__exit__)r r r r � __enter__O s
z EphemeralIPv4Network.__enter__c C s x| j D ]
}|� qW dS )z Teardown anything we set up.N)r )r � excp_type�
excp_value�excp_traceback�cmdr r r r& q s z EphemeralIPv4Network.__exit__c C s� dj | j| j�}tjd| j|| j� y| jjj | j|| j� W nD t
k
r� } z(dt |j �kr`� tjd| j| j� W Y dd}~X nPX | jjj
| jdd� | jjt| jjj| jdd�� | jjt| jjj| j|�� dS )z1Perform the ip comands to fully setup the device.z{0}/{1}z:Attempting setup of ephemeral network on %s with %s brd %sz File existsz7Skip ephemeral network setup, %s already has address %sNZinet)Zfamily)r r r r! r" r r r �net_opsZadd_addrr �str�stderr�link_upr �appendr Z link_downZdel_addr)r Zcidrr r r r r# v s.
z$EphemeralIPv4Network._bringup_devicec C sN xH| j D ]>\}}| jjj| j||� | jjdt| jjj| j||d�� qW d S )Nr )�gateway) r r r, Z append_router r �insertr � del_route)r Z net_addressr1 r r r r$ � s z+EphemeralIPv4Network._bringup_static_routesc C s� | j jj� }d|kr,tjd| j|j� � dS | j jj| j| j| j d� | j
j dt | j jj
| j| j| j d�� | j jj| jd| jd� | j
j dt | j jj
| jd�� dS )z<Perform the ip commands to fully setup the router if needed.�defaultz<Skip ephemeral route setup. %s already has default route: %sN)Zsource_addressr )r1 )r r, Zget_default_router! r" r �stripZ add_router r r r2 r r3 )r �outr r r r% � s*
z$EphemeralIPv4Network._bringup_router)NNN)�__name__�
__module__� __qualname__�__doc__r r r- r r r' r& r# r$ r% r r r r r s &"!r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �EphemeralIPv6Networkz�Context manager which sets up a ipv6 link local address
The linux kernel assigns link local addresses on link-up, which is
sufficient for link-local communication.
c C s" |st dj|���|| _|| _dS )z�Setup context manager and validate call signature.
@param interface: Name of the network interface to bring up.
@param ip: IP address to assign to the interface.
@param prefix: IPv6 uses prefixes, not netmasks
zCannot init network on {0}N)r r r r )r r r r r r r � s z EphemeralIPv6Network.__init__c C s& t j| jd�dkr"| jjj| j� dS )z�linux kernel does autoconfiguration even when autoconf=0
https://www.kernel.org/doc/html/latest/networking/ipv6.html
Z operstateZupN)r Z read_sys_netr r r, r/ )r r r r r' � s z EphemeralIPv6Network.__enter__c G s dS )z%No need to set the link to down stateNr )r �_argsr r r r& � s z EphemeralIPv6Network.__exit__N)r7 r8 r9 r: r r' r& r r r r r; � s
r; c @ sX e Zd Zdeeeef d�dd�Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dS )�EphemeralDHCPv4N)r
c C s( || _ d | _d | _|| _|| _|| _d S )N)�iface�_ephipv4�lease�
dhcp_log_funcr
r )r r r> r
rA r r r r � s zEphemeralDHCPv4.__init__c C s, | j r$tj| j �r$tjd| j � dS | j� S )zUSetup sandboxed dhcp context, unless connectivity_url can already be
reached.z:Skip ephemeral DHCP setup, instance has connectivity to %sN)r
r r r! r" � obtain_lease)r r r r r' � s zEphemeralDHCPv4.__enter__c C s | j � dS )z Teardown sandboxed dhcp context.N)�
clean_network)r r( r) r* r r r r& s zEphemeralDHCPv4.__exit__c C s d| _ | jr | jjddd� dS )z@Exit _ephipv4 context to teardown of ip configuration performed.N)r@ r? r& )r r r r rC s z EphemeralDHCPv4.clean_networkc C s� | j r | j S t| j| j| j�}|s(t� �|d | _ tjd| j d | j d | j d � ddddddgd d
�}| j|�}|d s�t j
|d |d
�|d < |d r�t j |d �|d< | j
r�| j
|d< t| jf|�}|j� || _| j S )a9 Perform dhcp discovery in a sandboxed environment if possible.
@return: A dict representing dhcp options on the most recent lease
obtained from the dhclient discovery if run, otherwise an error
is raised.
@raises: NoDHCPLeaseError if no leases could be obtained.
� z#Received dhcp lease on %s for %s/%sr z
fixed-addressz subnet-maskzbroadcast-addressz rfc3442-classless-static-routeszclassless-static-routesZrouters)r r r r r r r r r r r
���)r@ r
r r> rA r r! r" � extract_dhcp_options_mappingr Zmask_and_ipv4_to_bcast_addrr Zparse_static_routesr
r r' r? )r Zleases�nmap�kwargsZephipv4r r r rB s@
z EphemeralDHCPv4.obtain_leasec C sH i }x>|j � D ]2\}}t|t�r0| j|||� q| jj|�||< qW |S )N)�items�
isinstance�list�get_first_option_valuer@ �get)r rG �resultZinternal_reference�lease_option_namesr r r rF G s
z,EphemeralDHCPv4.extract_dhcp_options_mappingc C s, x&|D ] }|j |�s| jj |�||< qW d S )N)rM r@ )r Zinternal_mappingrO rN Zdifferent_namesr r r rL R s
z&EphemeralDHCPv4.get_first_option_value)NNN)r7 r8 r9 r r r- r r r' r& rC rB rF rL r r r r r= � s
2 r= c @ s2 e Zd ZdZd eed�dd�Zdd� Zd d
� Zd S )
�EphemeralIPNetworkah Combined ephemeral context manager for IPv4 and IPv6
Either ipv4 or ipv6 ephemeral network may fail to initialize, but if either
succeeds, then this context manager will not raise exception. This allows
either ipv4 or ipv6 ephemeral network to succeed, but requires that error
handling for networks unavailable be done within the context.
FT)�ipv6�ipv4c C s, || _ || _|| _tj� | _d| _|| _d S )N� )r rR rQ �
contextlib� ExitStack�stack� state_msgr )r r r rQ rR r r r r c s
zEphemeralIPNetwork.__init__c C s | j p
| js| S g }d}| j rzy | jjt| j| j�� d}W n< ttfk
rx } z t j
d| � |j |� W Y d d }~X nX | jr�y0| jjt | j| j�� d}|s�| j r�d| _
W n8 tk
r� } z t j
d| � |j |� W Y d d }~X nX |�st jd� |d �| S )NFTz Failed to bring up %s for ipv4.zusing link-local ipv6z Failed to bring up %s for ipv6.zGFailed to bring up EphemeralIPNetwork. Datasource setup cannot continuer )rR rQ rV �
enter_contextr= r r r r r! �infor0 r; rW �error)r �
exceptionsZephemeral_obtainedr r r r r' q s>
z EphemeralIPNetwork.__enter__c G s | j j� d S )N)rV �close)r r< r r r r& � s zEphemeralIPNetwork.__exit__N)FT)r7 r8 r9 r: �boolr r' r& r r r r rP Z s
,rP )r: rT Zlogging� functoolsr Ztypingr r r r r Z
cloudinit.netr Zcloudinit.net.dhcpr r r
Zcloudinit.subpr Z getLoggerr7 r! r r; r= rP r r r r �<module> s
; k