Current File : //snap/core20/2379/lib/python3/dist-packages/probert/__pycache__/network.cpython-38.pyc
U
T{)_Tk � @ s� d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z e� d�Z
zd dlmZmZ W n. ek
r� Z ze
�de� W 5 dZ[X Y nX dZdZd Zd
Zd Zd Zd
ZdZdZ dZ dZ dZ dZ dZ!dZ"dZ#ddddddd gZ$d d d d d!gZ%d"d#gZ&d$d%d&d'd(d)d*d+d,d-gd.d&d'd,d/id,d/id,d0id,d/id,d/id1�d2�d3�d,d/id&d'd,d4id,d4id5d,d/id,d6igid.d,d/id3�d5d/e$d7�d,d6igid5d/e%d7�d,d6igid5d/e&d7�d,d6igid8�d2�d&d9d&d5d,d/id,d6igid:�id5d,d/id,d6igid;�d&d,d0id,d0id,d0id,d0id,d/id<�d=�d&d'd,d4id,d4id.d,d/id3�d&d,d/id:�d>�d2�d&d'd,d6d/gid.d,d/id3�d,d6d/gid?�d2�d@�dA�Z'dBdC� Z(dDdE� Z)dFdG� Z*dHdI� Z+dJdK� Z,G dLdM� dM�Z-dNdOd%dPdQdR�Z.G dSdT� dT�Z/G dUdV� dVe j0�Z1G dWdX� dXe j0�Z2G dYdZ� dZe2�Z3d[d\� Z4d]d^� Z5ej6d_d`� �Z7G dadb� dbe1�Z8G dcdd� dd�Z9G dedf� df�Z:e;dgk�r�d dl<Z<d dl=Z=e8� Z>e>�?� Z@e<�<e>jA� e=�B� ZCe@D ]ZDeC�EeDe=jF� �q�eC�G� ZHeHD ]\ZDZe>�IeD� �q�e<�<e>jA� �q�dS )h� N)� OrderedDict)�udev_get_attributeszprobert.network)�_nl80211�
_rtnetlinkz)Failed import network library modules: %s� � � � � � �@ � � � i i i i i @ z
balance-rrz
active-backupz balance-xorZ broadcastz802.3adz balance-tlbz balance-albZlayer2zlayer2+3zlayer3+4zencap2+3zencap3+4ZslowZfastz'http://json-schema.org/draft-04/schema#�link�objectF� addresses�bond�bridge� netlink_data�type� udev_dataZarray�stringZinteger)�address�ip�family�source�scope)r �additionalProperties�
properties)r �itemsZbooleanZoneOfZnull)r �enum�Z is_masterZis_slave�masterZslaves�mode�xmit_hash_policy� lacp_rate�attrs)r r )r r r )�ifindex�flags�arptyper �name)r r �Z is_bridgeZis_portZ
interfaces�options)�ssid�
visible_ssids�
scan_state)r r r r r r �wlan)z$schema�titler r Zrequiredr c C s. | sdS t j�d| �}t j�|�s6t�d�|�� d S d}|dkr�d}t j�t j�|d��srt j�t j�|d��rxd}nxt j�t j�|d ��r�d }n\t j�t j�|d
��r�d }n@t j� t j�|d ��r�d
}n$t j�t j�d| ��r�| �
d�r�d}n�|dk�rd}n�|dk�rNt j�t j�|d
���r*d }n"t j�t j�|d���rHd}nd}n�|dk�r^d}n~|dk�rnd}nn|dk�r~d}n^|dk�r�d }nN|d k�r�d }n>|d k�r�d }n.|d!k�r�d"}n |d#k�r�d$}n|d%k�r�d&}| �
d'��s�| �
d(��r�d(}n| �
d)��r
d)}t |�d*k�r*t d+�| �� d S |S ),Nz???�/sys/class/netzNo sysfs path to {}r ZethZwirelessZphy80211r1 r Zbondingr Z tun_flagsZtapz/sys/devices/virtual/netZdummy� r Z create_childZibZibchildr Zpppi Zipipi Zip6tnli �loi Zsiti
Zgrei Zirdai! Zwlan_auxi�� ZtunZipppZisdnZmip6mnhar z)Failed to determine interface type for {})
�os�path�join�exists�log�debug�format�isdir�islink�isfile�
startswith�len�print)Zifacer* Z
sysfs_pathZDEV_TYPE� rC �1/usr/lib/python3/dist-packages/probert/network.py�
_compute_type� st � �
rE c sj �fdd�� �fdd��� �fdd�}��fdd�}� �fd d
�}� � �� |� |� |d �|d �|d
�d�S )Nc s t � t@ �dkS �Nr )�bool�
IFF_MASTERrC �r) rC rD �_iface_is_master s z&_get_bonding.<locals>._iface_is_masterc s t � t@ �dkS rF )rG � IFF_SLAVErC rI rC rD �_iface_is_slave s z%_get_bonding.<locals>._iface_is_slavec sH z*� � r"t d� ��� } | �� W S g W S W n tk
rB g Y S X d S )Nz /sys/class/net/%s/bonding/slaves��open�read�split�IOError)r �rJ �ifnamerC rD �_get_slave_iface_list s
z+_get_bonding.<locals>._get_slave_iface_listc sH z,� � r$t �d� �} t j�| �W S W d S W n tk
rB Y d S X d S )Nz/sys/class/net/%s/master)r6 �readlinkr7 �basenamerQ )r# )rL rS rC rD �_get_bond_master s
z&_get_bonding.<locals>._get_bond_masterc sd zH� � rFd�| f }t |��}|�� �� }W 5 Q R X |r@|d nd W S W n tk
r^ Y d S X d S )Nz /sys/class/net/%s/bonding/%sr rM )ZparamZ
bond_paramZbprR rC rD �_get_bond_param( s
z%_get_bonding.<locals>._get_bond_paramr$ r% r& r" rC )rS r) rT rW rX rC )rJ rL r) rS rD � _get_bonding s
�rY c sL �fdd�� �fdd��� �fdd�}� ��fdd�}� � �� |� |� d �S )
Nc s t j�d� d�} t j�| �S )Nr3 r �r6 r7 r8 r9 �� bridge_path�rS rC rD �_iface_is_bridge? s z'_get_bridging.<locals>._iface_is_bridgec s t j�d� d�} t j�| �S )Nr3 �brportrZ )Z bridge_portr] rC rD �_iface_is_bridge_portC s z,_get_bridging.<locals>._iface_is_bridge_portc s$ � � r t j�d�d�} t �| �S g S )Nr3 Zbrif)r6 r7 r8 �listdirr[ )r^ rS rC rD �_get_bridge_iface_listG s
z-_get_bridging.<locals>._get_bridge_iface_listc
s� t ddg�} � � r$tj�d�d�}n �� r<tj�d�d�}ni S i }t�|�D ]@}|| kr\qNtj�||�}t|��}|�� �� ||< W 5 Q R X qN|S )N�flushr r3 r_ )�setr6 r7 r8 ra rN rO �strip)Z
skip_attrsr\ r- Zbridge_attr_nameZbridge_attr_fileZ bridge_attr�r^ r` rS rC rD �_get_bridge_optionsM s
z*_get_bridging.<locals>._get_bridge_optionsr, rC )rS rb rg rC rf rD �
_get_bridging= s �rh c s � fdd�}t |�S )Nc s
| j � S �N)r )�obj��attrrC rD �getj s znetlink_attr.<locals>.get��property)rl rm rC rk rD � netlink_attri s rp c s � �fdd�}t |�S )Nc s& � D ] }|| j kr| j | S q�S ri �r )rj �k��keys�missingrC rD rm p s
zudev_attr.<locals>.getrn )rt ru rm rC rs rD � udev_attro s rv c @ s� e Zd Zedd� �Zedd� �Zd#dd�Zdd � Zd
d � Ze d �Z
e d
�Z e d�Z e
dd� �Zedddgd�Zedddgd�Zeddgd�Zed gd �Ze
d d� �Ze
d d� �Ze
d d� �Ze
d!d"� �ZdS )$�Linkc C sL t �|d �}|d r d}nt||d �}| i |||t||d �t|�d�S )Nr( Zis_vlanZvlanr* r) )r r r r r r )�socketZif_indextonamerE rY rh )�clsr r r+ �typrC rC rD �from_probe_dataz s �zLink.from_probe_datac C s< i }|d D ] }t �|�}||t|j�< q ||d< | f |�S )Nr )�Address�from_saved_data�strr )ry � link_dataZ address_objsZaddr�arC rC rD r} � s
zLink.from_saved_dataNc C s. || _ || _|| _|| _|| _|| _|| _d S ri )r r r r r r r1 )�selfr r r r r r r1 rC rC rD �__init__� s z
Link.__init__c C s | j d krg d d d�| _ d S )N)r/ r. r0 �r1 �r� rC rC rD � mark_as_wlan� s
�zLink.mark_as_wlanc C sP dd� | j �� D �| j| j| j| j| jd�}| jd k r@| j|d< t� |t
� |S )Nc S s g | ] }|� � �qS rC )� serialize)�.0r� rC rC rD �
<listcomp>� s z"Link.serialize.<locals>.<listcomp>)r r r r r r r1 ) r �valuesr r r r r r1 �
jsonschema�validate� link_schema)r� �rrC rC rD r� � s �
zLink.serializer) r( r+ c C s | j d d S �Nr' r rq r� rC rC rD �<lambda>� � z
Link.<lambda>ZID_VENDOR_FROM_DATABASEZ ID_VENDORZ ID_VENDOR_IDzUnknown VendorZID_MODEL_FROM_DATABASEZID_MODELZ ID_MODEL_IDz
Unknown ModelZ
ID_NET_DRIVERZ
ID_USB_DRIVERzUnknown DriverZDEVPATHzUnknown devpathc C s | j d d S r� rq r� rC rC rD r� � r� c C s | j t@ p| j t@ S ri )r) �IFF_UP� IFF_RUNNINGr� rC rC rD r� � s c C s | j �d�S )Nz/devices/virtual/)�devpathr@ r� rC rC rD r� � r� c C s | j r| j d S d S d S )Nr. r� r� rC rC rD r. � s
z Link.ssid)N)�__name__�
__module__� __qualname__� classmethodr{ r} r� r� r� rp r) r( r+ ro Zhwaddrrv ZvendorZmodelZdriverr� Z is_connectedZ
is_virtualr. rC rC rC rD rw x s:
�
��
�
�
�rw �globalZsiteZhostZnowhere)r �� � � � c @ s4 e Zd Zdd� Zdd� Zedd� �Zedd� �Zd S )
r| c C s, t �|�| _| jj| _|| _|| _|| _d S ri )� ipaddressZ ip_interfacer r r r r )r� r r r r rC rC rD r� � s
zAddress.__init__c C s | j | jt| j�| jd�S )N)r r r r )r r r~ r r r� rC rC rD r� � s
�zAddress.serializec C sV |d � d�}|d }|�dd�t@ r,d}nd}|d }tt�||��}| ||||�S ) N�local�latin-1r r) r ZstaticZdhcpr )�decoderm �IFA_F_PERMANENTr~ �
_scope_str)ry r r r r r rC rC rD r{ � s zAddress.from_probe_datac C s
t f |�S ri )r| )ry r rC rC rD r} � s zAddress.from_saved_dataN)r� r� r� r� r� r� r{ r} rC rC rC rD r| � s
r| c @ s, e Zd ZdZejdd� �Zejdd� �ZdS )�NetworkObserverzzA NetworkObserver observes the network state.
It calls methods on a NetworkEventReceiver in response to changes.
c C s d S ri rC r� rC rC rD �start s zNetworkObserver.startc C s d S ri rC �r� �fdrC rC rD �
data_ready
s zNetworkObserver.data_readyN)r� r� r� �__doc__�abc�abstractmethodr� r� rC rC rC rD r� s
r� c @ sH e Zd ZdZejdd� �Zejdd� �Zejdd� �Zejdd � �Z d
S ) �NetworkEventReceiverzTNetworkEventReceiver has methods called on it in response to network
changes.c C s d S ri rC �r� r( r rC rC rD �new_link s z NetworkEventReceiver.new_linkc C s d S ri rC �r� r( rC rC rD � update_link s z NetworkEventReceiver.update_linkc C s d S ri rC r� rC rC rD �del_link s z NetworkEventReceiver.del_linkc C s d S ri rC �r� �action�datarC rC rD � route_change s z!NetworkEventReceiver.route_changeN)
r� r� r� r� r� r� r� r� r� r� rC rC rC rD r� s
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TrivialEventReceiverc C s d S ri rC r� rC rC rD r� & s z TrivialEventReceiver.new_linkc C s d S ri rC r� rC rC rD r� ) s z TrivialEventReceiver.update_linkc C s d S ri rC r� rC rC rD r� , s z TrivialEventReceiver.del_linkc C s d S ri rC r� rC rC rD r� / s z!TrivialEventReceiver.route_changeN)r� r� r� r� r� r� r� rC rC rC rD r� $ s r� c s � fdd�}|S )Nc s � �fdd�}|S )Nc s� t �d� j||� � jf}�D ]}||| f7 }q || jkr�| j| \}}}|dkrf� ||f| j|< q�|dkr�� ||f| j|< q�|dkr�|dkr�| j|= q�� ||f| j|< n� ||f| j|< d S )Nzevent for %s: %s %s�NEW�CHANGE�DEL)r: r; r� �_calls)r� r� r� �keyrr Z prev_methZ prev_actionZ prev_data)�funcrt rC rD �wG s
z&coalesce.<locals>.decorator.<locals>.wrC �r� r� �rt �r� rD � decoratorF s zcoalesce.<locals>.decoratorrC )rt r� rC r� rD �coalesceC s r� c s � fdd�}|S )Nc s � ||f| j t� < d S ri )r� r r� r� rC rD r� i s znocoalesce.<locals>.wrC r� rC r� rD �
nocoalesceh s r� c c sB t � | _z
d V W 5 | j�� D ]\}}}|| ||� q d | _X d S ri )r r� r� )rj �methr� r� rC rC rD �CoalescedCallsn s
r� c @ sh e Zd ZdZddd�Zdd� Zdd� Zed �d
d � �Zed d �d
d� �Z e
dd� �Z dd� Z dd� Z
dS )� UdevObserverz,Use udev/netlink to observe network changes.Nc C s< i | _ t�� | _|d kr t� }t|t�s,t�|| _d | _ d S ri )
�_links�pyudevZContext�contextr� �
isinstancer� �AssertionError�receiverr� )r� r� rC rC rD r� | s
zUdevObserver.__init__c C s� t �| �| _t| �� | j�� W 5 Q R X | j�� | jji| _z4t�| �| _ | j �� | j�
| j �� | j ji� W n t k
r� t �
d� Y nX t| j�S )Nz could not start wlan_listener)r Zlistener�
rtlistenerr� r� �filenor� �_fdmapr �
wlan_listener�update� RuntimeErrorr: r; �listr� rC rC rD r� � s
�
�
zUdevObserver.startc C s$ t | �� | j| � W 5 Q R X d S ri )r� r� r� rC rC rD r� � s
zUdevObserver.data_readyr( c C s� t �d||� |�� D ]"\}}t|t�r|�dd�||< q|d }|dkrl|| jkrh| j|= | j�|� d S |dk�r|| jk�r
| j| }|j dkr�|j
t @ s�|d t @ r�z| � |� W n t
k
r� t �d � Y nX ||_|j d kr�t|j�|_ t|j|jd �|_| j�|� d S t| jjt|�d
��}t|�d k�rBd S |d }t|�} t|�| d < t � || �}
|
| j|< | j� ||
� d S )
Nzlink_change %s %s�utf-8�replacer( r� r� r1 r) zon-up trigger_scan failed)ZIFINDEXr r' ) r: r; r r� �bytesr� r� r� r� r r) r� � trigger_scanr� � exceptionr rE r+ rY r r� r� r� Z list_devicesr~ rA �dictr rw r{ r� ) r� r� r� rr �vr( ZdevZ udev_devicesZ udev_devicer r rC rC rD � link_change� sD
zUdevObserver.link_changer� c C s� t �d||� | j�|d �}|d kr*d S |d �d�}|dkrb|j�|d � | j�|d � d S t �
|�|j|< | j�|d � d S )Nzaddr_change %s %sr( r� r� r� ) r: r; r� rm r� r �popr� r� r| r{ )r� r� r� r r rC rC rD � addr_change� s zUdevObserver.addr_changec C sL t �d||� |�� D ]"\}}t|t�r|�dd�||< q| j�||� d S )Nzroute_change %s %sr� r� )r: r; r r� r� r� r� r� )r� r� r� rr r� rC rC rD r� � s
zUdevObserver.route_changec C s | j �|� d S ri )r� r� r� rC rC rD r� � s zUdevObserver.trigger_scanc C s� t �d|� |d }|dk s&|| jkr*d S | j| }|�� |d dkrRd|jd< |d dkr�d |kr�t� }|d D ]0\}}|�d
d �}|�|� |d krt||jd
< qtt|�|jd< d |jd< |d dk�r>|j t
@ �r
z| � |� W n t k
�r t �
d� Y nX n4z| j�|t
� W n t k
�r< t �
d� Y nX |d dk�sZ|d dk�r�t|�d g ��dk�r�|d d d �d
d �|jd
< |d dk�r�d |jd
< d S )Nz
wlan_event %sr( r �cmdZ TRIGGER_SCANZscanningr0 ZNEW_SCAN_RESULTS�ssidsr� r� z no statusr. r/ Z
NEW_INTERFACEzinitial trigger_scan failedzset_link_flags failedZ ASSOCIATEZ
DISCONNECT)r: r; r� r� r1 rd r� �add�sortedr) r� r� r� r� r� Zset_link_flagsrA rm )r� �argr( r r� r. ZstatusrC rC rD �
wlan_event� sB
�zUdevObserver.wlan_event)N)r� r� r� r� r� r� r� r� r� r� r� r� r� r� rC rC rC rD r� y s
,
r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �StoredDataObserverzVA cheaty observer that just pretends the network is in some
pre-arranged state.c C s, || _ | j d D ]}t�|t� q|| _d S )N�links)�
saved_datar� r� r� r� )r� r� r� r� rC rC rD r�
s zStoredDataObserver.__init__c C sL | j d D ] }t�|�}| j�|j|� q
| j d D ]}| j�d|� q4g S )Nr� �routesr� )r� rw r} r� r� r( r� )r� r� r rC rC rD r� s
zStoredDataObserver.startc C s d S ri rC r� rC rC rD r� s z StoredDataObserver.trigger_scanc C s d S ri rC r� rC rC rD r� s z StoredDataObserver.data_readyN)r� r� r� r� r� r� r� r� rC rC rC rD r� s
r� c @ s e Zd Zdd� ZdS )�
NetworkProberc C sl G dd� dt �}|� }t|�}|�� g g d�}|jD ]}|d �|�� � q6|jD ]}|d �|� qT|S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z/NetworkProber.probe.<locals>.CollectingReceiverc S s t � | _g | _d S ri )rd � all_links�
route_datar� rC rC rD r� & s z8NetworkProber.probe.<locals>.CollectingReceiver.__init__c S s | j �|� d S ri )r� r� r� rC rC rD r� * s z8NetworkProber.probe.<locals>.CollectingReceiver.new_linkc S s | j �|� d S ri )r� �appendr� rC rC rD r� - s z<NetworkProber.probe.<locals>.CollectingReceiver.route_changeN)r� r� r� r� r� r� rC rC rC rD �CollectingReceiver% s r� )r� r� r� r� )r� r� r� r� r� r� r� )r� r� Z collectorZobserverZresultsr r� rC rC rD �probe$ s
�
zNetworkProber.probeN)r� r� r� r� rC rC rC rD r� " s r� �__main__)Jr� � collectionsr �
contextlibr� r� Zloggingr6 rx r� Z
probert.utilsr Z getLoggerr: Zprobertr r � ImportError�eZwarningr� Z
IFF_BROADCASTZ IFF_DEBUGZ IFF_LOOPBACKZIFF_POINTOPOINTZIFF_NOTRAILERSr� Z IFF_NOARPZ IFF_PROMISCZ IFF_ALLMULTIrH rK Z
IFF_MULTICASTZ IFF_PORTSELZ
IFF_AUTOMEDIAr� Z
BOND_MODESZXMIT_HASH_POLICIESZ
LACP_RATESr� rE rY rh rp rv rw r� r| �ABCr� r� r� r� r� �contextmanagerr� r� r� r� r� ZpprintZselect�cr� Zfdsr� ZepollZpoll_obr� �registerZEPOLLINZpollZeventsr� rC rC rC rD �<module> sf
� � �
���� ������������%���� ����� ���
�
����yB1, ^� " %
Mini Shell