Current File : //snap/core18/2887/usr/lib/python3/dist-packages/cloudinit/net/__pycache__/ephemeral.cpython-36.pyc
3
�Ad�9 � @ s� d Z ddlZddlZddlmZmZmZmZ ddlj Z ddl
m Z ddl m
Z
mZmZ eje�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$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) �selfr r �prefix_or_maskr r r r �e� r �9/usr/lib/python3/dist-packages/cloudinit/net/ephemeral.py�__init__ s$ z EphemeralIPv4Network.__init__c C sR | j r(tj| j �r(tjd| j d � dS | j� | jr@| j� n| jrN| j � 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" x | j D ]}tj|dd� qW dS )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 z-family�inet�addr�delz%s/%s�devT)r$ N)r r )r Zaddressr r r r �_delete_addressn s
z$EphemeralIPv4Network._delete_addressc C s� dj | j| j�}tjd| j|| j� y0tjddddd|d| jd | jg
d
d d id
� W nF tjk
r� } z(dt |j
�krx� tjd| j| j� W Y dd}~X n^X tjdddddd | jdgd
d� | j j dddddd | jdg� | j 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 z-familyr* r+ �addr r- TZLANG�C)r$ Z
update_envz File existsz7Skip ephemeral network setup, %s already has address %sN�link�set�up)r$ Zdownr, )
r r r r r r r r ZProcessExecutionError�str�stderrr �append)r Zcidrr r r r r ~ sn z$EphemeralIPv4Network._bringup_devicec C sx xr| j D ]h\}}g }|dkr$d|g}tjdddd|g| d| jg dd � | jjd
dddd |g| d| jg � qW d S ) Nz0.0.0.0�viar z-4�router6 r- T)r$ r r, )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:tjd| j|j� � d S t j dd
dd | jd | jd
| jg dd� | jjddd
dd| jd | jd
| jg � t j dd
dd dd| jd | jg dd� | jjddd
dddd | jg� d S )z<Perform the ip commands to fully setup the router if needed.r r8 Zshowz 0.0.0.0/0T)r$ �defaultz<Skip ephemeral route setup. %s already has default route: %sNz-4r/ r- �srcr r, r7 ) r r r r �stripr r r r9 )r �out�_r r r r" � sT
z$EphemeralIPv4Network._bringup_router)NNN)�__name__�
__module__� __qualname__�__doc__r r r4 r r r# r) r. 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 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 s z EphemeralIPv6Network.__init__c C s2 t j| 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 operstater3 r r1 r2 r- F)r$ 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 �_argsr r r r) 4 s z EphemeralIPv6Network.__exit__N)r? r@ rA rB r r# r) r r r r rC s rC 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 �tmp_dir)r rF r rI rJ r r r r 9 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# G s zEphemeralDHCPv4.__enter__c C s | j � 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 | jjddd� dS )z@Exit _ephipv4 context to teardown of ip configuration performed.N)rH rG r) )r r r r rL X 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 |d �|d< | j r�| j |d< t
f |�}|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 ���)rH r rF rI rJ r r r � extract_dhcp_options_mappingr Zmask_and_ipv4_to_bcast_addrr r r
r# rG )r Zleases�nmap�kwargsZephipv4r r r rK ` 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_valuerH �get)r rP �resultZinternal_reference�lease_option_namesr r r rO � s
z,EphemeralDHCPv4.extract_dhcp_options_mappingc C s, x&|D ] }|j |�s| jj |�||< qW d S )N)rV rH )r Zinternal_mappingrX rW Zdifferent_namesr r r rU � s
z&EphemeralDHCPv4.get_first_option_value)NNNN)r? r@ rA r r r4 r r r# r) rL rK rO rU r r r r rE 8 s
2 rE c @ s2 e Zd ZdZd eed�dd�Zdd � Zd
d � ZdS )
�EphemeralIPNetworkz9Marries together IPv4 and IPv6 ephemeral context managersFTN)�ipv6�ipv4c C s, || _ || _|| _tj� | _d| _|| _d S )N� )r r[ rZ �
contextlib� ExitStack�stack� state_msgrJ )r r rZ r[ rJ r r r r � s
zEphemeralIPNetwork.__init__c C st y:| j r | jjt| j| jd�� | jr8| jjt| j�� W n4 tk
rn } z| jrZd| _ n|�W Y d d }~X nX | S )N)rJ zusing link-local ipv6)
r[ r_ �
enter_contextrE r rJ rZ rC r r` )r r r r r r# � s z EphemeralIPNetwork.__enter__c G s | j j� d S )N)r_ �close)r rD r r r r) � s zEphemeralIPNetwork.__exit__)FTN)r? r@ rA rB �boolr r# r) r r r r rY � s rY )rB r] ZloggingZtypingr r r r Z
cloudinit.netr Z cloudinitr Zcloudinit.net.dhcpr r r Z getLoggerr? r r
rC rE rY r r r r �<module> s
"m
Mini Shell