File: //usr/share/netplan/netplan/cli/__pycache__/utils.cpython-38.pyc
U
ci�_�' � @ s> d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl ZdZ
dZ G dd� dej �Z
e�ej�d��Zeje�e�e
��gej_dd� Zd d
� Zd d � Zd
d� Zdd� Zdd� Zd,dd�Z dg fdd�Z dd� Z dd� Z d d � Z d-d d �Z!d d!� Z"d"d#� Z#d$d%� Z$d&d'� Z%d(d)� Z&G d*d+� d+ej'�Z(dS ).� NzNetworkManager.servicez+snap.network-manager.networkmanager.servicec @ s( e Zd ZdejfdejfdejfgZdS )�_GErrorZdomain�code�messageN)�__name__�
__module__� __qualname__�ctypesZc_uint32Zc_int�c_char_pZ_fields_� r
r
�'/usr/share/netplan/netplan/cli/utils.pyr # s r �netplanc C sn t �� t�t�� }tt �| �� t�|���}|sDt |j
j � d���t �
t�|�� |rjt |j
j � d���dS )Nzutf-8T)�libZnetplan_clear_netdefsr �POINTERr �bool�netplan_parse_yaml�encodeZbyref� Exception�contentsr �decodeZnetplan_finish_parse)�path�errZretr
r
r �
netplan_parse+ s r c C s t j�dd�S )NZNETPLAN_GENERATE_PATHz/lib/netplan/generate)�os�environ�getr
r
r
r �get_generator_path8 s r c C s t jdddtgt jd�dkS )N� systemctl�--quietz
is-enabled)�stderrr )�
subprocess�call�NM_SNAP_SERVICE_NAME�DEVNULLr
r
r
r �is_nm_snap_enabled< s r# c C s, d}t � rd}tj|g| tjtjd� d S )N�nmcliznetwork-manager.nmcli��stdoutr )r# r �
check_callr" )�argsZ binary_namer
r
r r$ @ s r$ c C s2 zt dg� W dS ttjfk
r, Y dS X dS )z"Check if NetworkManager is runningZgeneralTFN)r$ �OSErrorr ZSubprocessErrorr
r
r
r �
nm_runningI s
r* c C sl t �d�}t� }| D ]R}t|d��>}|D ]2}|�|�}|r(|�tt�||�d���� q\q(W 5 Q R X q|S )Nz^interface-name=(.*)$�r� ) �re�compile�set�open�match�update�fnmatch�filter�group)�pathsZdevicesZpat�
interfacesr �f�line�mr
r
r �
nm_interfacesS s
r; Fc C s< t }d| g}|s|�d� t� r$t}|�|� t�|� d S )Nr �
--no-block)�NM_SERVICE_NAME�appendr# r! r r' )�action�syncZ service_name�commandr
r
r �systemctl_network_managera s
rB c C sB d| g}|s|� d� |� d� |D ]}|� |� q$t�|� d S )Nr r< zsystemd-networkd.service)r> r r' )r? r@ Zextra_servicesrA Zservicer
r
r �systemctl_networkdr s
rC c C s t �ddd| g�dkrdS dS )z4Return True if at least one matching unit is runningr r z is-activer TF)r r )Z unit_patternr
r
r �systemctl_is_active� s rD c C s t �ddg� dS )zEReload systemd unit files from disk and re-calculate its dependenciesr z
daemon-reloadN)r r' r
r
r
r �systemctl_daemon_reload� s rE c C s t jddd| gt jt jd� dS )z8Flush all IP addresses of a given interface via iproute2Zip�addr�flushr% N)r r' r" �Zifacer
r
r �
ip_addr_flush� s rI c
C s t j�d| �}|r�zTtt j�|d���8}|�� �� }|dkrZt�d| |� W 5 Q R � W d S W 5 Q R X W n: tk
r� } z t� d| t
|�� W Y �d S d }~X Y nX z(t j� t j�|dd��}t j� |�}W n> tk
�r } z t�d| |t
|�� W Y �d S d }~X Y nX |S ) Nz/sys/class/netZ operstateZdownz'device %s operstate is %s, not changingz$Cannot determine operstate of %s: %sZdevice�driverz0Cannot replug %s: cannot read link %s/device: %s)
r r �joinr0 �read�strip�logging�debug�IOError�error�str�realpath�basename)� interfaceZ only_downZdevdirr8 �state�erJ � driver_namer
r
r �get_interface_driver_name� s$ rY c C s t �| �t j d }|�d�S )Nr rF )� netifacesZ ifaddressesZAF_LINKr )rU �linkr
r
r �get_interface_macaddress� s r\ c C s t �| |�S �N)r3 � fnmatchcase)rU Z
match_namer
r
r �is_interface_matching_name� s r_ c C s t | �}t�||�S r] )rY r3 r^ )rU Z match_driverrX r
r
r �!is_interface_matching_driver_name� s r` c C s t | �}|�� |�� kS r] )r\ �lower)rU Z match_mac�
macaddressr
r
r � is_interface_matching_macaddress� s rc c s� t � t�st�� �dd�r$� �d�nd}t�| |�}t|�dkr`� �d�r`tt� fdd�|��}t|�dkr�� �d�r�tt� fd d�|��}t|�dkr�t� |� d S |d
S ) N�nameF�*r, rb c s t | � �d��S )Nrb )rc r rH �r1 r
r �<lambda>� � z%find_matching_iface.<locals>.<lambda>rJ c s t | � �d��S )NrJ )r` r rH rf r
r rg � rh r )
�
isinstance�dict�AssertionErrorr r3 r4 �len�listrN �info)r7 r1 Z name_globZmatchesr
rf r �find_matching_iface� s
ro c @ sF e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dS )�NetplanCommandTFc C s� || _ || _|| _|| _d | _d| _d | _i | _d | _d | _ t
j dt j
d |f |dd�| _| jjdddd � |s�| jjd
d d d
�| _| jjdddd�}|j| jd� d S )NFz%s %sr T)�prog� description�add_helpz--debug�
store_truezEnable debug messages)r? �helpzAvailable commands� �
subcommand)�title�metavar�destru zShow this help message)rr ru )�func)�
command_idrr � leaf_command�testing�_argsrO � commandclass� subcommandsrw r{ �argparse�ArgumentParser�sys�argv�parser� add_argumentZadd_subparsers�
subparsers�
add_parser� set_defaults� print_usage)�selfr| rr �leafr~ Zp_helpr
r
r �__init__� s6 �
� ��zNetplanCommand.__init__c C s
|| _ d S r] )r )r� r( r
r
r r2 � s zNetplanCommand.updatec C s> | j j| j| d�\}| _| js:| js:tdtjd� | �� d S )N)r( � namespacez You need to specify a command��file) r� Zparse_known_argsr rw r} �printr� r r� )r� �nsr
r
r �
parse_args� s zNetplanCommand.parse_argsc C s8 | j r| j �| j� | jr,d| jkr,| �� | �� d S )Nru )r� r2 r r} r� r{ �r� r
r
r � run_command� s
zNetplanCommand.run_commandc C s | j jtjd� t�tj� d S )Nr� )r� Z
print_helpr� r �exitr �EX_USAGEr� r
r
r r� s zNetplanCommand.print_usagec C s� |� }i | j |< || j | d< || j | d< |jrDtj�dd �sDd S | jj|j|j|jdd�}|j |j
|d� || j | d< d S )N�class�instanceZENABLE_TEST_COMMANDSF)rr ru rs )r{ r� r� ) r� r~ r r r r� r� r| rr r� �run)r� rd r� r� �pr
r
r �_add_subparser_from_class s
�z(NetplanCommand._add_subparser_from_classc C s@ dd l }|�|�D ](\}}|�|�rt|t�r| �||� qd S )Nr )�inspectZ
getmembersZisclass�
issubclassrp r� )r� Z
submodulesr� rd �objr
r
r �_import_subcommands s z"NetplanCommand._import_subcommandsN)TF)
r r r r� r2 r� r� r� r� r� r
r
r
r rp � s
rp )F)F))r� r rN r3 r� r rZ r- r Z ctypes.utilr= r! Z Structurer ZCDLLZutilZ find_libraryr
r r r Zargtypesr r r# r$ r* r; rB rC rD rE rI rY r\ r_ r` rc ro Z Namespacerp r
r
r
r �<module> s@