OwlCyberSecurity - MANAGER
Edit File: utils.cpython-38.pyc
U �p:ed' � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZdZdZ dd� Z dd� Z dd � Z e ed �d d �Zd d� Zdd� Zd0dd�Zd1dd�Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd2d"d#�Zd$d%� Zd&d'� Z d(d)� Z d*d+� Z d,d-� Z G d.d/� d/ej �Z!dS )3� NzNetworkManager.servicez+snap.network-manager.networkmanager.servicec C s t j�dd�S )NZNETPLAN_GENERATE_PATHz/lib/netplan/generate)�os�environ�get� r r �'/usr/share/netplan/netplan/cli/utils.py�get_generator_path 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 t jdg| t jt jd� d S )N�nmcli��stdoutr �r � check_callr ��argsr r r r ' s r )r �returnc C s t jdg| dd�S )Nr T�Zuniversal_newlines)r � check_outputr r r r � nmcli_out- 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_running3 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� interfaces�path�f�line�mr r r � nm_interfaces= s r. Fc C s" t � rt| tg|�S t| tg|�S �N)r r r �NM_SERVICE_NAME)�action�syncr r r �systemctl_network_managerK s r3 c C s: t |�dkr6d| g}|s"|�d� |�|� t�|� d S )Nr r z --no-block)�len�append�extendr r )r1 Zservicesr2 Zcommandr r r r S s r c C s` t � } tjdddgdd�}|�� D ]8}|�� �d�}|d �� r"|d d kr"| �|d � q"| S ) N� networkctlz --no-pagerz --no-legendTr � r ���)Z unmanagedZlinger)r! r r � splitlines�strip�split� isnumeric�add)r) �outr, �sr r r �networkd_interfaces_ s rA c C s t �ddg� d S )Nr7 Zreload�r r r r r r �networkctl_reloadi s rC c C s& t | �dkr"t�ddgt| � � d S )Nr r7 � reconfigure)r4 r r �list)r) r r r �networkctl_reconfigurem s rF 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_activer s rG c C s t �ddg� dS )zEReload systemd unit files from disk and re-calculate its dependenciesr z daemon-reloadNrB r r r r �systemctl_daemon_reloady s rH c C s t jddd| gt jt jd� dS )z8Flush all IP addresses of a given interface via iproute2Zip�addr�flushr Nr �Zifacer r r � ip_addr_flush~ s rL 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* �joinr" �readr; �logging�debug�IOError�error�str�realpath�basename)� interfaceZ only_downZdevdirr+ �state�erM � driver_namer r r �get_interface_driver_name� s$ r[ c C s& t �| ��t ji g�d }|�dd�S )Nr rI � )� netifacesZ ifaddressesr ZAF_LINK)rW �linkr r r �get_interface_macaddress� s r_ c C s t �| |�S r/ �r% Z fnmatchcase)rW Z match_namer r r �is_interface_matching_name� s ra c s2 |}t |t�r|g}t| �� t� fdd�|D ��S )Nc 3 s | ]}t �� |�V qd S r/ r` )�.0�pattern�rZ r r � <genexpr>� s �z4is_interface_matching_driver_name.<locals>.<genexpr>)� isinstancerT r[ �any)rW Z match_driverZ driver_globsr rd r �!is_interface_matching_driver_name� s �rh c C s t | �}|�� |�� kS r/ )r_ �lower)rW Z match_mac� macaddressr r r � is_interface_matching_macaddress� s rk 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 rj c s t | � �d��S )Nrj )rk r rK �r# r r �<lambda>� � z%find_matching_iface.<locals>.<lambda>rM c s t | � �d��S )NrM )rh r rK rn r r ro � rp r ) rf �dict�AssertionErrorr r% r& r4 rE rP �info)r) r# Z name_globZmatchesr rn r �find_matching_iface� s rt 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)r1 �helpzAvailable commandsr\ � subcommand)�title�metavar�destrz zShow this help message)rw rz )�func)� command_idrw � leaf_command�testing�_argsrQ � commandclass� subcommandsr{ r �argparse�ArgumentParser�sys�argv�parser� add_argumentZadd_subparsers� subparsers� add_parser� set_defaults� print_usage)�selfr� rw �leafr� Zp_helpr r r �__init__� s6 � � ��zNetplanCommand.__init__c C s || _ d S r/ )r� )r� r r r r r$ � 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� r{ 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 )Nrz )r� r$ 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)rw rz rx )r r� r� ) r� r� r r r r� r� r� rw r� �run)r� rl 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� issubclassru r� )r� Z submodulesr� rl �objr r r �_import_subcommands s z"NetplanCommand._import_subcommandsN)TF) �__name__� __module__� __qualname__r� r$ r� r� r� r� r� r r r r ru � s ru )F)F)F)"r� r rP r� r r] r% r r0 r r r r rE rT r r r. r3 r rA rC rF rG rH rL r[ r_ ra rh rk rt Z Namespaceru r r r r �<module> s<