Your IP : 216.73.217.13
U
�Ad�9 � @ s� d Z ddlZddlZddlmZmZmZmZ ddlm Z ddl
m Z ddl m
Z
mZmZ e�e�ZG dd� d�ZG dd � d �ZG d
d � d �ZG d d
� d
�ZdS )z.Module for ephemeral network context managers
� N)�Any�Dict�List�Optional)�subp)�NoDHCPLeaseError� maybe_perform_dhcp_discovery�parse_static_routesc @ s\ e Zd ZdZdeeeef d�dd�Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� ZdS )�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$td�||||���zt�|�| _W n2 tk
rf } ztd�|��|�W 5 d}~X Y 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) �selfr r �prefix_or_maskr r r r �e� r �9/usr/lib/python3/dist-packages/cloudinit/net/ephemeral.py�__init__ s4 �����z EphemeralIPv4Network.__init__c C sR | j r(t�| j �r(t�d| j d � dS | �� | jr@| �� n| jrN| � � 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_router�r r r r � __enter__L s �
z EphemeralIPv4Network.__enter__c C s | j D ]}tj|dd� qdS )z Teardown anything we set up.T��captureN)r r )r � excp_type�
excp_value�excp_traceback�cmdr r r �__exit__i s
z EphemeralIPv4Network.__exit__c C s, t j dddddd||f d| jgdd � d
S ) z7Perform the ip command to remove the specified address.r �-family�inet�addr�delz%s/%s�devTr& N)r r )r Zaddressr r r r �_delete_addressn s
�
�z$EphemeralIPv4Network._delete_addressc C s� d� | j| j�}t�d| j|| j� z0tjddddd|d| jd | jg
d
d d id
� W nF tjk
r� } z&dt |j
�krx� t�d| j| j� W 5 d}~X Y n^X tjdddddd | jdgd
d� | j � dddddd | jdg� | j � ddddd|d | jg� dS )z1Perform the ip comands to fully setup the device.z{0}/{1}z:Attempting setup of ephemeral network on %s with %s brd %sr r- r. r/ �addr r1 TZLANG�C)r' Z
update_envz File existsz7Skip ephemeral network setup, %s already has address %sN�link�set�upr& Zdownr0 )
r r r r r r r r ZProcessExecutionError�str�stderrr �append)r Zcidrr r r r r! ~ s� �� �
��
�
�� ��z$EphemeralIPv4Network._bringup_devicec C st | j D ]h\}}g }|dkr"d|g}tjdddd|g| d| jg dd � | j�d
dddd |g| d| jg � qd S ) Nz0.0.0.0�viar �-4�router: r1 Tr& r r0 )r r r r �insert)r Z net_addressZgatewayZvia_argr r r r" � s( ��� ���z+EphemeralIPv4Network._bringup_static_routesc C s� t j ddddgdd�\}}d|kr:t�d| j|�� � d S t j dd
dd | jd | jd
| jg dd� | j�ddd
dd| jd | jd
| jg � t j dd
dd dd| jd | jg dd� | j�ddd
dddd | jg� d S )z<Perform the ip commands to fully setup the router if needed.r r= Zshowz 0.0.0.0/0Tr& �defaultz<Skip ephemeral route setup. %s already has default route: %sNr< r3 r1 �srcr r0 r; ) r r r r �stripr r r r> )r �out�_r r r r# � sf �� ���� � �z$EphemeralIPv4Network._bringup_router)NNN)�__name__�
__module__� __qualname__�__doc__r r r8 r r r% r, r2 r! r" r# r r r r r
s �� - Jr
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 d�|���|| _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 s z EphemeralIPv6Network.__init__c C s2 t �| jd�dkr.tjdddd| jdgdd� d S )
z�linux kernel does autoconfiguration even when autoconf=0
https://www.kernel.org/doc/html/latest/networking/ipv6.html
Z operstater7 r r5 r6 r1 Fr& N)r Z read_sys_netr 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 Z_argsr r r r, 4 s z EphemeralIPv6Network.__exit__N)rD rE rF rG r r% r, r r r r rH s rH c @ sX e Zd Zdeeeef d�dd�Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dS )�EphemeralDHCPv4Nr c C s( || _ d | _d | _|| _|| _|| _d S �N)�iface�_ephipv4�lease�
dhcp_log_funcr �tmp_dir)r rL r rO rP r r r r 9 s zEphemeralDHCPv4.__init__c C s, | j r$t�| j �r$t�d| j � dS | �� 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_leaser$ r r r r% G s �zEphemeralDHCPv4.__enter__c C s | � � dS )z Teardown sandboxed dhcp context.N)�
clean_network)r r( r) r* r r r r, T s zEphemeralDHCPv4.__exit__c C s* | j r d| _ | jsdS | j�ddd� dS )z@Exit _ephipv4 context to teardown of ip configuration performed.N)rN rM r, r$ r r r rR X s
z EphemeralDHCPv4.clean_networkc C s� | j r | j S t| j| j| j�}|s(t� �|d | _ t�d| j d | j d | j d � ddddddgd d
�}| �|�}|d s�t �
|d |d
�|d < |d r�t |d �|d< | j r�| j |d< t
f |�}|�� || _| 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 )rN r rL rO rP r r r � extract_dhcp_options_mappingr Zmask_and_ipv4_to_bcast_addrr r r
r% rM )r Zleases�nmap�kwargsZephipv4r r r rQ ` sR �
���
��
z EphemeralDHCPv4.obtain_leasec C sD i }|� � D ]2\}}t|t�r.| �|||� q | j�|�||< q |S rK )�items�
isinstance�list�get_first_option_valuerN �get)r rU �resultZinternal_reference�lease_option_namesr r r rT � s
�z,EphemeralDHCPv4.extract_dhcp_options_mappingc C s( |D ] }|� |�s| j� |�||< qd S rK )r[ rN )r Zinternal_mappingr] r\ Zdifferent_namesr r r rZ � s
z&EphemeralDHCPv4.get_first_option_value)NNNN)rD rE rF r r r8 r r r% r, rR rQ rT rZ r r r r rJ 8 s ��
2 rJ c @ s2 e Zd ZdZd eed�dd�Zdd � Zd
d � ZdS )
�EphemeralIPNetworkz9Marries together IPv4 and IPv6 ephemeral context managersFTN)�ipv6�ipv4c C s, || _ || _|| _t�� | _d| _|| _d S )N� )r r` r_ �
contextlib� ExitStack�stack� state_msgrP )r r r_ r` rP r r r r � s
zEphemeralIPNetwork.__init__c
C st z:| j r | j�t| j| jd�� | jr8| j�t| j�� W n4 tk
rn } z| jrZd| _ n|�W 5 d }~X Y nX | S )N)rP zusing link-local ipv6)
r` rd �
enter_contextrJ r rP r_ rH r re )r r r r r r% � s �z EphemeralIPNetwork.__enter__c G s | j �� d S rK )rd �closerI r r r r, � s zEphemeralIPNetwork.__exit__)FTN)rD rE rF rG �boolr r% r, r r r r r^ � s �� r^ )rG rb Zlogging�typingr r r r Z
cloudinit.netr Z cloudinitr Zcloudinit.net.dhcpr r r Z getLoggerrD r r
rH rJ r^ r r r r �<module> s
"m