/usr/lib/python3/dist-packages/cloudinit/net/__pycache__
Edit: /usr/lib/python3/dist-packages/cloudinit/net/__pycache__/ephemeral.cpython-310.pyc (16568B)
o
th,V @ s d Z ddlZddlZddlmZ ddlmZmZmZm Z m
Z
m Z m Z ddl
mZ ddlmZ ddlmZmZ ddlmZ ddlmZmZ eeZG dd d Z G d
d d Z G d d
d
Z G dd dZ de ee ef de e fddZ!dS )z-Module for ephemeral network context managers N)partial)AnyCallableDictListLiteralOptionalTuple)NoDHCPLeaseError maybe_perform_dhcp_discovery)ProcessExecutionError)UrlError wait_for_urlc @ sL e Zd ZdZ ddefddZdd Zdd Zd
d Zd 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_urls_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.
Ninterface_addrs_before_dhcpc
C s t ||||gstd||||zt|| _W n ty. } ztd| | d} ~ ww || _|| _|| _|| _ || _
g | _ || _ | j d| j | _
|| ji | _dS )a 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 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
ValueErrorformatnetipv4_mask_to_net_prefixprefix interfaceip broadcastrouter
static_routes cleanup_cmdsdistrocidrgetr )
selfr r r prefix_or_maskr r r r e r$ 9/usr/lib/python3/dist-packages/cloudinit/net/ephemeral.py__init__ s8
z EphemeralIPv4Network.__init__c
C s z?z| W n! ty( } zdt|jvr dt|jvr W Y d}~nd}~ww | jr3| W dS | jr=| W dS W dS tyN | ddd w )zSet up ephemeral network if interface is not connected.
This context manager handles the lifecycle of the network interface,
addresses, routes, etc
z File existszAddress already assignedN) _bringup_devicer strstderrr _bringup_static_routesr _bringup_router__exit__)r! r# r$ r$ r% __enter__M s( z EphemeralIPv4Network.__enter__c C s | j D ]}| qdS )z Teardown anything we set up.N)r )r! excp_type
excp_valueexcp_tracebackcmdr$ r$ r% r, q s
z EphemeralIPv4Network.__exit__c C s: t d| j| j| j t | ji }|d}| jd}| j dd |di D v }| j dd | jdi D v }|rHt d| j| j n | j
j | j| j| j |r^t d| j n
| j
j j
| jd d
|rrt d | j n| jt| j
j j| jd d
|rt d | j d
S | jt| j
j j| j| j d
S )af Perform the ip commands to fully set up the device.
Dhcp clients behave differently in how they leave link state and ip
address assignment.
Attempt assigning address and setting up link if needed to be done.
Set cleanup_cmds to return the interface state to how it was prior
to execution of the dhcp client.
z:Attempting setup of ephemeral network on %s with %s brd %supc S g | ]}| d qS r r .0r r$ r$ r%
s
z8EphemeralIPv4Network._bringup_device..ipv4c S r3 r4 r5 r6 r$ r$ r% r8 s z1Skip adding ip address: %s already has address %sz9Skip bringing up network link: interface %s is already upinet)familyzLNot queueing link down: link [%s] was up prior before receiving a dhcp leasezSNot queueing address removal: address %s was assigned before receiving a dhcp leaseN)LOGdebugr r r netinfo netdev_infor r r r net_opsadd_addrlink_upr appendr link_downdel_addr)r! interface_addrs_after_dhcphas_linkhad_linkhas_iphad_ipr$ r$ r% r' v sl
z$EphemeralIPv4Network._bringup_devicec
C sJ | j D ] \}}| jj| j|| | jdt| jjj| j||d qd S )Nr gateway) r r r@ append_router r insertr del_route)r! net_addressrL r$ r$ r% r* s z+EphemeralIPv4Network._bringup_static_routesc C s | j j }d|v rtd| j| dS | j jj| j| j| j d | j
dt | j jj
| j| j| j d | j jj| jd| jd | j
dt | j jj
| jd dS )z r? r )r! r rf rc rd r$ r$ r% r& s
zEphemeralDHCPv4.__init__c C s& t | j }rtd| dS | S )zUSetup sandboxed dhcp context, unless connectivity_url can already be
reached.z:Skip ephemeral DHCP setup, instance has connectivity to %sN)_check_connectivity_to_imdsrc r< r= obtain_lease)r! imds_reached_at_urlr$ r$ r% r- ' s zEphemeralDHCPv4.__enter__c C s | dS )z Teardown sandboxed dhcp context.N)
clean_network)r! r. r/ r0 r$ r$ r% r, 6 s zEphemeralDHCPv4.__exit__c C s$ d| _ | jr| jddd dS dS )z@Exit _ephipv4 context to teardown of ip configuration performed.N)rh rg r, r_ r$ r$ r% rl : s z EphemeralDHCPv4.clean_networkc C s | j r| j S t| j| j| j| _ | j st td| j d | j d | j d ddddg ddd}| |}|d sGt
|d
|d |d < |d rV| jj |d |d < t
| jfd
| ji|}| || _| 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-address)z rfc3442-classless-static-routeszclassless-static-routesr z unknown-121routers)r r r" r r r r r" r r r )rh r r rf rd r
r< r= extract_dhcp_options_mappingr mask_and_ipv4_to_bcast_addr dhcp_clientparse_static_routesr r r- rg )r! nmapkwargsephipv4r$ r$ r% rj @ sP
z EphemeralDHCPv4.obtain_leasec C sD i }| D ]\}}t|tr| ||| q| j|||< q|S re )items
isinstancelistget_first_option_valuerh r )r! rr resultinternal_referencelease_option_namesr$ r$ r% rn w s
z,EphemeralDHCPv4.extract_dhcp_options_mappingc C s( |D ]}| |s| j |||< qd S re )r rh )r! internal_mappingr{ ry different_namesr$ r$ r% rx s
z&EphemeralDHCPv4.get_first_option_value)NNN)rW rX rY r r r r( r r r& r- r, rl rj rn rx r$ r$ r$ r% rb s
7 rb c @ sp e Zd ZdZ ddededeeeee f fdd Z
d
d Z d e d
de
eee f fddZdd ZdS )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.
FTNipv6r9 rc c C s2 || _ || _|| _t | _d| _|| _|| _dS )a
Args:
distro: The distro object
interface: The interface to bring up
ipv6: Whether to bring up an ipv6 network
ipv4: Whether to bring up an ipv4 network
connectivity_urls_data: List of url data to use for connectivity
check before attempting to bring up ephemeral networks. If
connectivity can be established to any of the urls, then the
ephemeral network setup is skipped.
N) r r9 r
contextlib ExitStackstack state_msgr rc )r! r r r r9 rc r$ r$ r% r& s
zEphemeralIPNetwork.__init__c C s | j s| js| S g }d}t| j }rtd| | S td | j r6| jdd\}}||O }|r6|| | jrL| jdd\}}||O }|rL|| | jrQ|sT| j sWd| _|sbt d |d | S )
NFz:We already have connectivity to IMDS at %s, skipping DHCP.z/No connectivity to IMDS, attempting DHCP setup.r9 )
ip_versionr zusing link-local ipv6zGFailed to bring up EphemeralIPNetwork. Datasource setup cannot continuer )
r9 r ri rc r< r= _perform_ephemeral_network_setuprC r error)r!
exceptionsephemeral_obtainedimds_urlipv4_ephemeral_obtainedipv4_exceptionipv6_ephemeral_obtainedipv6_exceptionr$ r$ r% r- sB
z EphemeralIPNetwork.__enter__r )r9 r returnc
C s z3|dkr| j t| j| jd n|dkr"| j t| j| j ntd| td| j| W dS t t
fyS } ztd| j| d|fW Y d }~S d }~ww )
a
Attempt to bring up an ephemeral network for the specified IP version.
Args:
ip_version (str): The IP version to bring up ("ipv4" or "ipv6").
Returns:
Tuple: A tuple containing:
- a boolean indicating whether an ephemeral network was
successfully obtained
- an optional exception if ephemeral network setup failed
or None if successful
r9 )r rf r zUnsupported IP version: z7Successfully brought up %s for ephemeral %s networking.)TNz2Failed to bring up %s for ephemeral %s networking.FN) r
enter_contextrb r r r\ r r< r= r r
)r! r r# r$ r$ r% r s> z3EphemeralIPNetwork._perform_ephemeral_network_setupc G s | j d S re )r closer` r$ r$ r% r, s zEphemeralIPNetwork.__exit__)FTN)rW rX rY rZ boolr r r r( r r& r- r r Exceptionr r, r$ r$ r$ r% r~ s&
4
1r~ rc r c
s fdd} st d dS tdd D pd}ztdd D ||d
d d \}}W n tyE } z
t d
| W Y d}~dS d}~ww |sOt d dS |S )a
Perform a connectivity check to the provided URLs to determine if the
ephemeral network setup is necessary.
This function attempts to reach one of the provided URLs and returns the
URL that was successfully reached. If none of the URLs can be reached,
it returns None.
The timeout for the request is determined by the highest timeout value
provided in the connectivity URLs data. If no timeout is provided, a
default timeout of 5 seconds is used.
Args:
connectivity_urls_data: A list of dictionaries, each containing
the following keys:
- "url" (str): The URL to check connectivity for.
- "headers" (dict, optional): Headers to include in the request.
- "timeout" (int, optional): Timeout for the request in seconds.
Returns:
Optional[str]: The URL that was successfully reached, or None if no
connectivity was established.
c s fddD d }|S )z
Helper function to get headers for a given URL from the connectivity
URLs data provided to _check_connectivity_to_imds.
c s" g | ]
}|d kr| dqS urlheadersr5 r7 url_datar r$ r% r8 7 s
zD_check_connectivity_to_imds.._headers_cb..r r$ r rc r r% _headers_cb2 s
z0_check_connectivity_to_imds.._headers_cbzZNo connectivity URLs provided. Skipping connectivity check before ephemeral network setup.Nc s s | ] }| d dV qdS )timeoutr Nr5 r r$ r$ r% G s z._check_connectivity_to_imds.. c S s g | ]}|d qS r r$ r r$ r$ r% r8 M s z/_check_connectivity_to_imds..Fr )urls
headers_cbr connect_synchronouslymax_waitz8Failed to reach IMDS without ephemeral network setup: %sz5Failed to reach IMDS without ephemeral network setup.)r< r= maxr r
)rc r r url_that_worked_r# r$ r r% ri s:
ri )"rZ r logging functoolsr typingr r r r r r r
cloudinit.netr cloudinit.netinfor> cloudinit.net.dhcpr
r cloudinit.subpr cloudinit.url_helperr
r getLoggerrW r< r r\ rb r~ r( ri r$ r$ r$ r% s, $
f s