آگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآ 3 \¼me!م@s\ddlZddlZddlZddlmZmZmZddlmZej e ƒZdd„ZGdd„dƒZ dS)éN)عnetعsubpعutil)عipv4_mask_to_net_prefixcCs`|jdƒ}|jdƒ}g}x2ttdƒƒD]"}|jt||ƒt||ƒ@ƒq&Wdjdd„|DƒƒS)z” Return the subnet for a given ip address and a netmask @return (str): the subnet @param ip: ip address @param netmask: netmask ع.écSsg|]}t|ƒ‘qS©)عstr)ع.0عxrrْ /usr/lib/python3.6/config_nic.pyْ szgen_subnet..)عsplitعlistعrangeعappendعintعjoin)عipعnetmaskZip_arrayZ mask_arrayعresultعindexrrrع gen_subnets "rc@sbeZdZddd„Zdd„Zdd„Zdd „Zd d„Zdd „Zdd„Z ddd„Z dd„Zddd„ZdS)عNicConfiguratorTcCsV||_i|_d|_d|_|r&|jƒn"x |jD]}|j|j|jjƒ<q.W|jƒ|_ dS)a Initialize the Nic Configurator @param nics (list) an array of nics to configure @param use_system_devices (Bool) Get the MAC names from the system if this is True. If False, then mac names will be retrieved from the specified nics. N) عnicsعmac2Nameعipv4PrimaryGatewayZipv6PrimaryGatewayعfind_devicesعnameعmacعlowerعget_primary_nicع_primaryNic)عselfrZuse_system_devicesعnicrrrع__init__#s zNicConfigurator.__init__cCsFdd„|jDƒ}|sdSt|ƒdkr:tddd„|Dƒƒ‚n|dSdS)z| Retrieve the primary nic if it exists @return (NicBase): the primary nic if exists, None otherwise cSsg|]}|jr|‘qSr)عprimary)r r$rrrr =sz3NicConfigurator.get_primary_nic..Néz!There can only be one primary niccSsg|] }|j‘qSr)r)r r$rrrr Csr)rعlenعRuntimeError)r#Zprimary_nicsrrrr!8szNicConfigurator.get_primary_nicc Cs€dddg}tj|ƒ\}}tjdd|ƒdd…}d}xF|D]>}tj||ƒ}|sPq:|jdƒjƒ}|jd dƒd } | |j|<q:WdS)zb Create the mac2Name dictionary The mac address(es) are in the lower case rعaddrZshowz\n\d+: ع r'Nz3link/ether (([0-9A-Fa-f]{2}[:]){5}([0-9A-Fa-f]{2}))ْ:r)rعrerعsearchعgroupr r) r#عcmdعoutputZ_errZsectionsZmacPatZsectionعmatchrrrrrrHs zNicConfigurator.find_devicesc Cs²|jjƒ}|jj|ƒ}|s(td|jƒ‚g}d||dœ}g}g}|j||ƒ\}} |j|ƒ|j| ƒ|j||ƒ\}} |j|ƒ|j| ƒ|jd|iƒ|j |ƒ|r®|j|ƒ|S)z¾ Return the config list needed to configure a nic @return (list): the subnets and routes list to configure the nic @param nic (NicBase): the nic to configure zNo known device has MACADDR: %sZphysical)عtyperZmac_addressعsubnets) rr rعgetع ValueErrorعgen_ipv4عextendعgen_ipv6عupdater) r#r$rrع nics_cfg_listZcfgعsubnet_listع route_listr4عroutesrrrعgen_one_nicZs& zNicConfigurator.gen_one_niccCsi}g}|jr|jddiƒ|jjƒ}|jjƒdkr8d}|dkrX|jddiƒ|g|fS|jddiƒ|j}|sz|g|fS|d}|jrک|jd |jiƒ|jr®|jd |jiƒ|jrà|j rà|j d|_ |jd|j iƒ|g|fS|joى|j گr|jd|j||j |jƒiƒ|g|fS) a+ Return the set of subnets and routes needed to configure the IPv4 settings of a nic @return (set): the set of subnet and routes to configure the gateways @param name (str): subnet and route list for the nic @param nic (NicBase): the nic to configure ZcontrolعautoZdisabledZmanualZstaticr3عdhcprعaddressrعgatewayr>) Zonbootr:Z bootProtor Z ipv4_modeZ staticIpv4rrr&عgatewaysrr"عgen_ipv4_route)r#rr$عsubnetr=Z bootprotoعaddrsZv4rrrr7~s6 zNicConfigurator.gen_ipv4cCsDg}t|ƒ}x2|D]*}dt||ƒ|f}|j|d|ddœƒqW|S)zے Return the routes list needed to configure additional Ipv4 route @return (list): the route list to configure the gateways @param nic (NicBase): the nic to configure @param gateways (str list): the list of gateways z%s/%dZroutei')عdestinationr3rCZmetric)rrr)r#r$rDrr=ZcidrrCrHrrrrE°s zNicConfigurator.gen_ipv4_routecCsL|jsggfSg}|j}x&|D]}d|j|jdœ}|j|ƒqWg}||fS)a Return the set of subnets and routes needed to configure the gateways for a nic @return (set): the set of subnets and routes to configure the gateways @param name (str): name of the nic @param nic (NicBase): the nic to configure Zstatic6)r3rBr)Z staticIpv6rrr)r#rr$r<rGr*rFr=rrrr9بs zNicConfigurator.gen_ipv6FNcCs@|rtjdƒ|j|ƒg}x|jD]}|j|j|ƒƒq$W|S)z@Return the config elements that are needed to configure the nicszConfiguring the interfaces file)عloggerعinfoع configurerr8r?)r#rKعosfamilyr;r$rrrعgenerateés zNicConfigurator.generatecCstjdƒtjjjƒdS)NzClearing DHCP leases)rIrJrrAZIscDhclientZclear_leases)r#rrrع clear_dhcpِs zNicConfigurator.clear_dhcpcCsŒ|s|dkrtjdƒdSd}tjj|dƒ}tjj|dƒ}tjj|ƒrbtjj|ƒrbtj||ƒddd g}tj|d j|ƒdچ|j ƒdS)z^ Configure the /etc/network/interfaces Make a back up of the original Zdebianz3Debian OS not detected. Skipping the configure stepNz/etc/networkZ interfacesz&interfaces.before_vmware_customizationzH# DO NOT EDIT THIS FILE BY HAND -- AUTOMATICALLY GENERATED BY cloud-initz"source /etc/network/interfaces.d/*z*source-directory /etc/network/interfaces.dr+)Zcontent) rIrJعosعpathrعexistsعrenamerZ write_filerN)r#rLZ containingDirZ interfaceFileZoriginalFileعlinesrrrrKْs zNicConfigurator.configure)T)FN)N) ع__name__ع __module__ع__qualname__r%r!rr?r7rEr9rMrNrKrrrrr"s $2! r)ZloggingrOr-Z cloudinitrrrZcloudinit.net.network_staterZ getLoggerrTrIrrrrrrعs