관리-도구
편집 파일: config_nic.cpython-39.pyc
a '�Dg! � @ s\ d dl Z d dlZd dlZd dlmZmZmZ d dlmZ e � e �Z dd� Z G dd� d�Z dS )� N)�net�subp�util)�ipv4_mask_to_net_prefixc C s\ | � d�}|� d�}g }ttd��D ]"}|�t|| �t|| �@ � q$d�dd� |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)�.0�xr r �S/usr/lib/python3.9/site-packages/cloudinit/sources/helpers/vmware/imc/config_nic.py� <listcomp> � z gen_subnet.<locals>.<listcomp>)�split�list�range�append�int�join)�ip�netmaskZip_arrayZ mask_array�result�indexr r r � gen_subnet s r c @ sb e Zd Zddd�Zdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z ddd�Z dd� Z ddd�Z dS )�NicConfiguratorTc C sR || _ i | _d| _d| _|r&| �� n | j D ]}|j| j|j�� < q,| �� | _ 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)�selfr Zuse_system_devices�nicr 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 = r z3NicConfigurator.get_primary_nic.<locals>.<listcomp>N� z!There can only be one primary nicc S s g | ] }|j �qS r )r r( r r r r C r r )r �len� RuntimeError)r$ Z primary_nicsr r r r" 8 s �z NicConfigurator.get_primary_nicc C sz g d�}t � |�\}}t�dd| �dd� }d}|D ]>}t�||�}|sLq6|�d��� }|�dd�d } | | j|< q6dS ) 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_errZsectionsZmacPat�section�matchr r r r r r H s z NicConfigurator.find_devicesc C s� |j �� }| j�|�}|s(td|j ��g }d||d�}g }g }| �||�\}} |�|� |�| � | �||�\}} |�|� |�| � |�d|i� |� |� |r�|�|� |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_address�subnets) r r! r �get� ValueError�gen_ipv4�extend�gen_ipv6�updater ) r$ r% r r � nics_cfg_listZcfg� subnet_list� route_listr7 �routesr r r � gen_one_nicZ s& zNicConfigurator.gen_one_nicc C s i }g }|j r |�ddi� |j�� }|j�� dkr8d}|dkrX|�ddi� |g|fS |�ddi� |j}|sz|g|fS |d }|jr�|�d |ji� |jr�|�d |ji� |jr�|j r�|j d | _ |�d | j i� |g|fS | j �s |j �r |�d | � ||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�autoZdisabledZmanualZstaticr6 �dhcpr �addressr �gatewayrA ) Zonbootr= Z bootProtor! Z ipv4_modeZ staticIpv4r r r' �gatewaysr r# �gen_ipv4_route)r$ r r% �subnetr@ Z bootproto�addrsZv4r r r r: ~ s8 �zNicConfigurator.gen_ipv4c C s@ g }t |�}|D ]*}dt||�|f }|�|d|dd�� q|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' )� destinationr6 rF Zmetric)r r r )r$ r% rG r r@ ZcidrrF rK r r r rH � s �� z NicConfigurator.gen_ipv4_routec C sH |j sg g fS g }|j }|D ] }d|j|jd�}|�|� q 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)r6 rE r )Z staticIpv6r r r )r$ r r% r? rJ r, rI r@ r r r r< � s � zNicConfigurator.gen_ipv6FNc C s< |rt �d� | �|� g }| jD ]}|�| �|�� q"|S )z@Return the config elements that are needed to configure the nicsz Configuring the interfaces file)�logger�info� configurer r; rB )r$ rN �osfamilyr> r% r r r �generate� s zNicConfigurator.generatec C s t �d� tjj�� d S )NzClearing DHCP leases)rL rM r rD Z IscDhclientZ clear_leases)r$ r r r � clear_dhcp� s zNicConfigurator.clear_dhcpc C s� |r |dkrt �d� dS d}tj�|d�}tj�|d�}tj�|�s^tj�|�r^t�||� g d�}tj|d�|�d � | � � 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_customization)zH# 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) rL rM �os�pathr �exists�renamer Z write_filerQ )r$ rO Z containingDirZ interfaceFileZ originalFile�linesr r r rN � s � zNicConfigurator.configure)T)FN)N) �__name__� __module__� __qualname__r& r" r rB r: rH r< rP rQ rN r r r r r " s $2! r )ZloggingrR r/ Z cloudinitr r r Zcloudinit.net.network_stater Z getLoggerrW rL r r r r r r �<module> s