U Ld\5@sdZddlZddlZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddlmZmZmZeeZGdddZGd d d ZGd d d ZGd ddZdS)z.Module for ephemeral network context managers N)partial)AnyCallableDictListOptional)subp) IscDhclientNoDHCPLeaseErrormaybe_perform_dhcp_discoveryc@sTeZdZdZdeeeefdddZddZ dd Z d d Z d d Z ddZ dS)EphemeralIPv4NetworkaContext 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. Nconnectivity_url_datac Cst||||gs$td||||zt||_Wn2tk rf} ztd| | W5d} ~ XYnX||_||_||_||_ ||_ ||_ g|_ ||_ dS)aXSetup 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) selfrrrprefix_or_maskrrrrer9/usr/lib/python3/dist-packages/cloudinit/net/ephemeral.py__init__ s6zEphemeralIPv4Network.__init__cCs|jr(t|jr(td|jddSz*||jrB|n|jrP| Wn0t j k rt d| dddYnXdS)z>Perform ephemeral network setup if interface is not connected.z=Skip ephemeral network setup, instance has connectivity to %sZurlNzHError bringing up EphemeralIPv4Network. Datasource setup cannot continue)rrhas_url_connectivityLOGdebug_bringup_devicer_bringup_static_routesr_bringup_routerrProcessExecutionErrorerror__exit__rrrr __enter__Os&   zEphemeralIPv4Network.__enter__cCs|jD] }|qdS)zTeardown anything we set up.N)r)r excp_type excp_valueexcp_tracebackcmdrrrr)us zEphemeralIPv4Network.__exit__c Csd|j|j}td|j||jz|jj |j||jWnFt j k r}z&dt |j krbtd|j|jW5d}~XYnPX|jjj|jdd|jt|jjj|jdd|jt|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)rrrr"r#rrrnet_opsZadd_addrrr'strstderrlink_uprappendrZ link_downZdel_addr)rZcidrrrrrr$zs8z$EphemeralIPv4Network._bringup_devicec CsJ|jD]>\}}|jj|j|||jdt|jjj|j||dqdS)Nrgateway) rrr0Z append_routerrinsertr del_route)rZ net_addressr6rrrr%sz+EphemeralIPv4Network._bringup_static_routesc Cs|jj}d|kr,td|j|dS|jjj|j|j|j d|j dt |jjj |j|j|j d|jjj|jd|jd|j dt |jjj |jddS)z  z$EphemeralIPv4Network._bringup_router)NNN)__name__ __module__ __qualname____doc__rrr1rr r+r)r$r%r&rrrrr s /&!r c@s(eZdZdZddZddZddZdS) EphemeralIPv6NetworkzContext 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. cCs"|std|||_||_dS)zSetup 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)rrrr)rrrrrrr szEphemeralIPv6Network.__init__cCs&t|jddkr"|jj|jdS)zlinux kernel does autoconfiguration even when autoconf=0 https://www.kernel.org/doc/html/latest/networking/ipv6.html Z operstateZupN)rZ read_sys_netrrr0r3r*rrrr+szEphemeralIPv6Network.__enter__cGsdS)z%No need to set the link to down stateNrrZ_argsrrrr)szEphemeralIPv6Network.__exit__N)r<r=r>r?r r+r)rrrrr@s r@c@sXeZdZdeeeefdddZddZddZ d d Z d d Z d dZ ddZ dS)EphemeralDHCPv4Nr cCs(||_d|_d|_||_||_||_dSN)iface_ephipv4lease dhcp_log_funcrr)rrrDrrGrrrr s zEphemeralDHCPv4.__init__cCs,|jr$t|jr$td|jdS|S)zUSetup sandboxed dhcp context, unless connectivity_url can already be reached.z:Skip ephemeral DHCP setup, instance has connectivity to %sN)rrr!r"r# obtain_leaser*rrrr+s zEphemeralDHCPv4.__enter__cCs |dS)z Teardown sandboxed dhcp context.N) clean_network)rr,r-r.rrrr)szEphemeralDHCPv4.__exit__cCs*|jr d|_|jsdS|jddddS)z@Exit _ephipv4 context to teardown of ip configuration performed.N)rFrEr)r*rrrrIs zEphemeralDHCPv4.clean_networkcCs|jr |jSt|j|j|j}|s(t|d|_td|jd|jd|jdddddddgd d }||}|d st |d |d |d <|drt |d|d<|j r|j |d<t|jf|}|||_|jS)a9Perform 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/%srz fixed-addressz subnet-maskzbroadcast-addresszrfc3442-classless-static-routeszclassless-static-routesZrouters)rrrrrrrrrrr)rFr rrDrGr r"r#extract_dhcp_options_mappingrZmask_and_ipv4_to_bcast_addrr Zparse_static_routesrr r+rE)rZleasesnmapkwargsZephipv4rrrrHsR    zEphemeralDHCPv4.obtain_leasecCsDi}|D]2\}}t|tr.||||q |j|||<q |SrC)items isinstancelistget_first_option_valuerFget)rrLresultZinternal_referencelease_option_namesrrrrKMs z,EphemeralDHCPv4.extract_dhcp_options_mappingcCs(|D]}||s|j|||<qdSrC)rRrF)rZinternal_mappingrTrSZdifferent_namesrrrrQXs z&EphemeralDHCPv4.get_first_option_value)NNN)r<r=r>rrr1rr r+r)rIrHrKrQrrrrrBs  2 rBc@s2eZdZdZd eedddZddZd d Zd S) EphemeralIPNetworkz9Marries together IPv4 and IPv6 ephemeral context managersFT)ipv6ipv4cCs,||_||_||_t|_d|_||_dS)N)rrWrV contextlib ExitStackstack state_msgr)rrrrVrWrrrr cs  zEphemeralIPNetwork.__init__c Csvz<|jr|jt|j|j|jr:|jt|j|jWn4tk rp}z|jr\d|_ n|W5d}~XYnX|S)Nzusing link-local ipv6) rWr[ enter_contextrBrrrVr@r r\)rrrrrr+qs  zEphemeralIPNetwork.__enter__cGs|jdSrC)r[closerArrrr)szEphemeralIPNetwork.__exit__N)FT)r<r=r>r?boolr r+r)rrrrrU`s rU)r?rYZlogging functoolsrtypingrrrrrZ cloudinit.netrZ cloudinitrZcloudinit.net.dhcpr r r Z getLoggerr<r"r r@rBrUrrrrs    ? m