3
\me! @ s\ d dl Z d dlZd dlZd dlmZmZmZ d dlmZ e j e
Z dd Z G dd dZ
dS ) N)netsubputil)ipv4_mask_to_net_prefixc C s` | j d}|j d}g }x2ttdD ]"}|jt|| t|| @ q&W djdd |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
. c S s g | ] }t |qS )str).0xr r /usr/lib/python3.6/config_nic.py
s z gen_subnet..)splitlistrangeappendintjoin)ipnetmaskZip_arrayZ
mask_arrayresultindexr r r
gen_subnet s
"r c @ sb e Zd ZdddZdd Zdd Zdd Zd
d Zd d
Zdd Z dddZ
dd Z dddZ dS )NicConfiguratorTc C sV || _ i | _d| _d| _|r&| j n"x | j D ]}|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)
nicsmac2Nameipv4PrimaryGatewayZipv6PrimaryGateway find_devicesnamemaclowerget_primary_nic _primaryNic)selfr Zuse_system_devicesnicr r r __init__# s
zNicConfigurator.__init__c C sF dd | j D }|sdS t|dkr:tddd |D n|d S dS )z|
Retrieve the primary nic if it exists
@return (NicBase): the primary nic if exists, None otherwise
c S s g | ]}|j r|qS r )primary)r
r$ r r r r
= s z3NicConfigurator.get_primary_nic..N z!There can only be one primary nicc S s g | ]
}|j qS r )r )r
r$ r r r r
C s r )r len RuntimeError)r# Z primary_nicsr r r r! 8 s z NicConfigurator.get_primary_nicc
C s dddg}t j |\}}tjdd| dd }d}xF|D ]>}tj||}|sPq:|jdj }|jd dd
} | | j|< q:W dS ) 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 searchgroupr r )
r# cmdoutputZ_errZsectionsZmacPatZsectionmatchr r r r r r H s
z NicConfigurator.find_devicesc
C s |j j }| 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
z No known device has MACADDR: %sZphysical)typer Z mac_addresssubnets)
r r r get
ValueErrorgen_ipv4extendgen_ipv6updater )
r# r$ r r
nics_cfg_listZcfg subnet_list
route_listr4 routesr r r gen_one_nicZ s&
zNicConfigurator.gen_one_nicc C s i }g }|j r |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 | j o|j r |jd | j ||j |ji |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
ZcontrolautoZdisabledZmanualZstaticr3 dhcpr addressr gatewayr> )
Zonbootr: Z bootProtor Z ipv4_modeZ
staticIpv4r r r& gatewaysr r" gen_ipv4_route)r# r r$ subnetr= Z bootprotoaddrsZv4r r r r7 ~ s6
zNicConfigurator.gen_ipv4c C sD g }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' ) destinationr3 rC Zmetric)r r r )r# r$ rD r r= ZcidrrC rH r r r rE s
z NicConfigurator.gen_ipv4_routec C sL |j sg g fS g }|j }x&|D ] }d|j|jd}|j| q W g }||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)r3 rB r )Z
staticIpv6r r r )r# r r$ r< rG r* rF r= r r r r9 s
zNicConfigurator.gen_ipv6FNc C s@ |rt jd | j| g }x | jD ]}|j| j| q$W |S )z@Return the config elements that are needed to configure the nicsz Configuring the interfaces file)loggerinfo configurer r8 r? )r# rK osfamilyr; r$ r r r generate s
zNicConfigurator.generatec C s t jd tjjj d S )NzClearing DHCP leases)rI rJ r rA Z IscDhclientZ clear_leases)r# r r r
clear_dhcp s
zNicConfigurator.clear_dhcpc C s | s|dkr t jd dS d}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)
rI rJ ospathr existsrenamer Z
write_filerN )r# rL Z
containingDirZ
interfaceFileZ originalFilelinesr r r rK s
zNicConfigurator.configure)T)FN)N)
__name__
__module__ __qualname__r% r! r r? r7 rE r9 rM rN rK r r r r r " s
$2!
r )ZloggingrO r- Z cloudinitr r r Zcloudinit.net.network_stater Z getLoggerrT rI r r r r r r s