File: //lib/python3/dist-packages/cloudinit/net/__pycache__/ephemeral.cpython-310.pyc
o
�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 @ sb e Zd ZdZ ddeeeef fdd�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�std�||||���zt�|�| _W n ty. } ztd�|��|�d}~ww || _|| _|| _|| _ || _
|| _ 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�net�ipv4_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__ s0 ������
z EphemeralIPv4Network.__init__c C sX | j rt�| j �rt�d| j d � dS | �� | jr!| �� dS | jr*| � � dS dS )z>Perform ephemeral network setup if interface is not connected.z=Skip ephemeral network setup, instance has connectivity to %s�urlN)
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 �addressr r r r �_delete_addressn s
�
�z$EphemeralIPv4Network._delete_addressc C s d� | j| j�}t�d| j|| j� ztjddddd|d| jd | jg
d
d d id
� W n% tjyO } zdt |j
�vr;� t�d| j| j� W Y d}~dS d}~ww 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/ r0 �addr r2 T�LANG�C)r( �
update_envz File existsz7Skip ephemeral network setup, %s already has address %sN�link�set�upr' �downr1 )
r r r r r! r r r �ProcessExecutionError�str�stderrr �append)r �cidrr r r r r" ~ s� ��
����
�
�
�� ��z$EphemeralIPv4Network._bringup_devicec C st | j D ]4\}}g }|dkrd|g}tjdddd|g| d| jg dd � | j�d
dddd |g| d| jg � qd S ) Nz0.0.0.0�viar �-4�router@ r2 Tr' r r1 )r r r r �insert)r � net_address�gateway�via_argr r r r# � s* ��� ����z+EphemeralIPv4Network._bringup_static_routesc C s� t j g d�dd�\}}d|v rt�d| j|�� � dS t j ddd d
| jd | jd | jg dd� | j�d
ddd d| jd | jd | jg � t j ddd d
dd| jd | jg dd� | j�d
ddd ddd | jg� dS )z<Perform the ip commands to fully setup the router if needed.)r rD �showz 0.0.0.0/0Tr' �defaultz<Skip ephemeral route setup. %s already has default route: %sNr rC rD r5 r2 �srcr r1 rB ) r r r! r �stripr r r rE )r �out�_r r r r$ � sd �� ���� ��z$EphemeralIPv4Network._bringup_router)NNN)�__name__�
__module__� __qualname__�__doc__r r r> r r r& r- r4 r" r# r$ r r r r r
s �
�- J 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 |s t 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 s6 t �| jd�dkrtjdddd| jdgdd� d S d S )
z�linux kernel does autoconfiguration even when autoconf=0
https://www.kernel.org/doc/html/latest/networking/ipv6.html
� operstater; r r9 r: r2 Fr' N)r � 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)rO rP rQ rR r r& r- r r r r rS s
rS c @ s` e Zd Z ddeeeef fdd�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 rZ r r] r^ r r r r 9 s
zEphemeralDHCPv4.__init__c C s, | j rt�| j �rt�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 rd| _ | js dS | j�ddd� dS )z@Exit _ephipv4 context to teardown of ip configuration performed.N)r\ r[ r- r% r r r r` X s
z EphemeralDHCPv4.clean_networkc C s� | j r| j S t| j| j| j�}|st� �|d | _ t�d| j d | j d | j d � ddddddgd d
�}| �|�}|d sJt �
|d |d
�|d < |d rVt |d �|d< | j r^| j |d< t
di |��}|�� || _| 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-routes�routers)r r r r r r r r r r r Nr )r\ r rZ r] r^ r r r! � extract_dhcp_options_mappingr �mask_and_ipv4_to_bcast_addrr r r
r& r[ )r �leases�nmap�kwargs�ephipv4r r r r_ ` sL �
���
��
z EphemeralDHCPv4.obtain_leasec C sD i }|� � D ]\}}t|t�r| �|||� q| j�|�||< q|S rY )�items�
isinstance�list�get_first_option_valuer\ �get)r rf �result�internal_reference�lease_option_namesr r r rc � s
�z,EphemeralDHCPv4.extract_dhcp_options_mappingc C s( |D ]}|� |�s| j� |�||< qd S rY )rm r\ )r �internal_mappingrp rn �different_namesr r r rl � s
��z&EphemeralDHCPv4.get_first_option_value)NNNN)rO rP rQ r r r> r r r& r- r` r_ rc rl r r r r rX 8 s �
�
2 rX c @ s: e Zd ZdZ d
dedefdd�Zd d
� Zd d � ZdS )�EphemeralIPNetworkz9Marries together IPv4 and IPv6 ephemeral context managersFTN�ipv6�ipv4c C s, || _ || _|| _t�� | _d| _|| _d S )N� )r ru rt �
contextlib� ExitStack�stack� state_msgr^ )r r rt ru r^ r r r r � s
zEphemeralIPNetwork.__init__c
C s| z!| j r| j�t| j| jd�� | jr | j�t| j�� W | S W | S ty= } z| jr0d| _ n|�W Y d }~| S d }~ww )N)r^ zusing link-local ipv6)
ru ry �
enter_contextrX r r^ rt rS r rz )r r r r r r& � s$ �� �
���z EphemeralIPNetwork.__enter__c G s | j �� d S rY )ry �closerV r r r r- � s zEphemeralIPNetwork.__exit__)FTN)rO rP rQ rR �boolr r& r- r r r r rs � s ��
� rs )rR rw �logging�typingr r r r �
cloudinit.netr � cloudinitr �cloudinit.net.dhcpr r r � getLoggerrO r r
rS rX rs r r r r �<module> s
"m