U Ad9@sdZddlZddlZddlmZmZmZmZddlm Z ddl m Z ddl m Z mZmZeeZGdddZGdd d ZGd d d ZGd d d ZdS)z.Module for ephemeral network context managers N)AnyDictListOptional)subp)NoDHCPLeaseErrormaybe_perform_dhcp_discoveryparse_static_routesc@s\eZdZdZdeeeefdddZddZ dd Z d d Z d d Z ddZ ddZdS)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_cmds) selfrrprefix_or_maskrrr rer9/usr/lib/python3/dist-packages/cloudinit/net/ephemeral.py__init__s4zEphemeralIPv4Network.__init__cCsR|jr(t|jr(td|jddS||jr@|n|jrN| dS)z>Perform ephemeral network setup if interface is not connected.z=Skip ephemeral network setup, instance has connectivity to %sZurlN) r rhas_url_connectivityLOGdebug_bringup_devicer_bringup_static_routesr_bringup_routerrrrr __enter__Ls   zEphemeralIPv4Network.__enter__cCs|jD]}tj|ddqdS)zTeardown anything we set up.TcaptureN)rr)r excp_type excp_valueexcp_tracebackcmdrrr__exit__is zEphemeralIPv4Network.__exit__c Cs,tjdddddd||fd|jgdd d S) z7Perform the ip command to remove the specified address.r-familyinetaddrdelz%s/%sdevTr&N)rr)rZaddressrrrr_delete_addressns  z$EphemeralIPv4Network._delete_addressc Csd|j|j}td|j||jz0tjddddd|d|jd |jg d d d id WnFtjk r}z&dt |j krxtd|j|jW5d}~XYn^Xtjdddddd |jdgd d|j dddddd |jdg|j ddddd|d |jgdS)z1Perform the ip comands to fully setup the device.z{0}/{1}z:Attempting setup of ephemeral network on %s with %s brd %srr-r.r/addrr1TZLANGC)r'Z update_envz File existsz7Skip ephemeral network setup, %s already has address %sNlinksetupr&Zdownr0) rrrrr rrrZProcessExecutionErrorstrstderrrappend)rZcidrrrrrr!~s     z$EphemeralIPv4Network._bringup_devicec Cst|jD]h\}}g}|dkr"d|g}tjdddd|g|d|jgdd |jd dddd |g|d|jgqdS) Nz0.0.0.0viar-4router:r1Tr&rr0)rrrrinsert)rZ net_addressZgatewayZvia_argrrrr"s(  z+EphemeralIPv4Network._bringup_static_routesc Cstjddddgdd\}}d|kr:td|j|d Stjdd dd |jd |jd |jg dd|jddd dd|jd |jd |jg tjdd dd dd|jd |jg dd|jddd dddd |jgd S)z)rout_rrrr#sf  z$EphemeralIPv4Network._bringup_router)NNN)__name__ __module__ __qualname____doc__rrr8rrr%r,r2r!r"r#rrrrr s -Jr 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)rrr)rrrrrrszEphemeralIPv6Network.__init__cCs2t|jddkr.tjdddd|jdgddd S) zlinux kernel does autoconfiguration even when autoconf=0 https://www.kernel.org/doc/html/latest/networking/ipv6.html Z operstater7rr5r6r1Fr&N)rZ read_sys_netrrr$rrrr%)s zEphemeralIPv6Network.__enter__cGsdS)z%No need to set the link to down stateNrrZ_argsrrrr,4szEphemeralIPv6Network.__exit__N)rDrErFrGrr%r,rrrrrHs  rHc@sXeZdZdeeeefdddZddZddZ d d Z d d Z d dZ ddZ dS)EphemeralDHCPv4Nr cCs(||_d|_d|_||_||_||_dSN)iface_ephipv4lease dhcp_log_funcr tmp_dir)rrLr rOrPrrrr9s 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)r rrrr obtain_leaser$rrrr%Gs zEphemeralDHCPv4.__enter__cCs |dS)z Teardown sandboxed dhcp context.N) clean_network)rr(r)r*rrrr,TszEphemeralDHCPv4.__exit__cCs*|jr d|_|jsdS|jddddS)z@Exit _ephipv4 context to teardown of ip configuration performed.N)rNrMr,r$rrrrRXs 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 f|}|||_|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 )rNrrLrOrPrrr extract_dhcp_options_mappingrZmask_and_ipv4_to_bcast_addrr r r r%rM)rZleasesnmapkwargsZephipv4rrrrQ`sR     zEphemeralDHCPv4.obtain_leasecCsDi}|D]2\}}t|tr.||||q |j|||<q |SrK)items isinstancelistget_first_option_valuerNget)rrUresultZinternal_referencelease_option_namesrrrrTs z,EphemeralDHCPv4.extract_dhcp_options_mappingcCs(|D]}||s|j|||<qdSrK)r[rN)rZinternal_mappingr]r\Zdifferent_namesrrrrZs z&EphemeralDHCPv4.get_first_option_value)NNNN)rDrErFrrr8rrr%r,rRrQrTrZrrrrrJ8s  2 rJc@s2eZdZdZd eedddZdd Zd d ZdS) EphemeralIPNetworkz9Marries together IPv4 and IPv6 ephemeral context managersFTN)ipv6ipv4cCs,||_||_||_t|_d|_||_dS)N)rr`r_ contextlib ExitStackstack state_msgrP)rrr_r`rPrrrrs  zEphemeralIPNetwork.__init__c Cstz:|jr |jt|j|jd|jr8|jt|jWn4tk rn}z|jrZd|_ n|W5d}~XYnX|S)N)rPzusing link-local ipv6) r`rd enter_contextrJrrPr_rHrre)rrrrrr%szEphemeralIPNetwork.__enter__cGs|jdSrK)rdcloserIrrrr,szEphemeralIPNetwork.__exit__)FTN)rDrErFrGboolrr%r,rrrrr^s r^)rGrbZloggingtypingrrrrZ cloudinit.netrZ cloudinitrZcloudinit.net.dhcprrr Z getLoggerrDrr rHrJr^rrrrs   "m