3
\meS9 @ s d Z ddlZddlZddlmZ ddlmZmZmZm Z m
Z
ddl j Z ddl
mZmZmZ ddlmZ ejeZG dd dZG d d
d
ZG d d d ZG d
d dZdS )z.Module for ephemeral network context managers
N)partial)AnyCallableDictListOptional) IscDhclientNoDHCPLeaseError maybe_perform_dhcp_discovery)ProcessExecutionErrorc @ sT e Zd ZdZdeeeef dddZdd 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 ||||gs$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
ValueErrorformatnetZipv4_mask_to_net_prefixprefixr
interfaceip broadcastrouter
static_routes cleanup_cmdsdistro)
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_connectivityLOGdebug_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_valueexcp_tracebackcmdr 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 strstderrlink_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 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 st j
|d |d
|d < |d rt 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 ZleasesnmapkwargsZephipv4r r r rB s@
z EphemeralDHCPv4.obtain_leasec C sH i }x>|j D ]2\}}t|tr0| j||| q| jj|||< qW |S )N)items
isinstancelistget_first_option_valuer@ get)r rG resultZinternal_referencelease_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 eedddZdd 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)ipv6ipv4c C s, || _ || _|| _tj | _d| _|| _d S )N )r rR rQ
contextlib ExitStackstack 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 | jry0| jjt | j| j d}|s| j rd| _
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 s
; k