File: //opt/alt/python37/lib/python3.7/site-packages/pyroute2/ethtool/__pycache__/ethtool.cpython-37.pyc
B
��f�= � @ sb d dl Z d dlmZ d dlmZmZ d dlmZmZm Z m
Z
m Z m Z m
Z
d dlmZmZ d dlmZ d dlmZmZ ed�jZed�jZe �e�Zed d
�Z G d d � d e �Z G d
d� de �Z G dd� de �Z!G dd� dedd��Z"G dd� dedd��Z#G dd� dedd��Z$G dd � d ed d ��Z%G d d � d ed d ��Z&G d!d"� d"ed"d#��Z'G d$d%� d%�Z(dS )&� N)�
namedtuple)�c_uint16�c_uint32)�LINK_DUPLEX_NAMES�LINK_PORT_NAMES�LINK_TP_MDI_NAMES�LINK_TRANSCEIVER_NAMES�LinkModeBits_by_index� LMBTypeMode� LMBTypePort)�
WAKE_NAMES� IoctlEthtool)� NetlinkError)� NlEthtool�ethtool_rings_msg����EthtoolBitsetBit)�index�name�enable�setc @ s e Zd ZdS )�UseIoctlN)�__name__�
__module__� __qualname__� r r �I/opt/alt/python37/lib/python3.7/site-packages/pyroute2/ethtool/ethtool.pyr s r c @ s$ e Zd Zedd� �Zedd� �ZdS )�EthtoolCoalescec C s dd� | � � D �S )Nc S s i | ]\}}t |�|�qS r )�int)�.0r �valuer r r �
<dictcomp># s z.EthtoolCoalesce.from_ioctl.<locals>.<dictcomp>)�items)�ioctl_coalescer r r �
from_ioctl! s zEthtoolCoalesce.from_ioctlc C s. x(|� � D ] \}}| | |kr
|| |< q
W d S )N)r" )r# �coalescer r r r r �to_ioctl% s zEthtoolCoalesce.to_ioctlN)r r r � staticmethodr$ r& r r r r r s r c @ s e Zd ZdZdd� ZdS )�EthtoolFeature)r r r r � availablec C s" || _ || _|| _|| _|| _d S )N)r r r r r) )�selfr r r r r) r r r �__init__/ s
zEthtoolFeature.__init__N)r r r � __slots__r+ r r r r r( , s r( c @ s$ e Zd Zedd� �Zedd� �ZdS )�EthtoolFeaturesc C s | dd� |D ��S )Nc S s( i | ] \}}}}}t |||||�|�qS r )r( )r r r r) r r r r r r! ; s z.EthtoolFeatures.from_ioctl.<locals>.<dictcomp>r )�cls�featuresr r r r$ 8 s zEthtoolFeatures.from_ioctlc C s: x4|j �� D ]&}| |j }|j|kr&q |j| |j< q W d S )N)r/ �valuesr r )�ioctl_featuresZ eth_features�featurer r r r r& A s
zEthtoolFeatures.to_ioctlN)r r r � classmethodr$ r' r& r r r r r- 7 s r- )r/ c @ s$ e Zd Zedd� �Zedd� �ZdS )�EthtoolWakeOnLanc C s| |d � d�}i }x\|� d�d D ]J}|d }|� d�}|� d�}|� d�}t|||d kr^d nd
d d �||< q"W t|d d �S )
Nr ZETHTOOL_A_WOL_MODES�ETHTOOL_A_BITSET_BITS�attrs� �ETHTOOL_A_BITSET_BIT_INDEX�ETHTOOL_A_BITSET_BIT_NAME�ETHTOOL_A_BITSET_BIT_VALUETF)r )�modes�sopass)�get_attrr r4 )r. Znl_wolZ wol_modes�moder r r r r r � from_netlinkK s
"z EthtoolWakeOnLan.from_netlinkc C sN i }x<t �� D ]0\}}|j|@ rt|||j|@ dkd d�||< qW t|d d�S )Nr )r )r; r< )r r" � supportedr Zwoloptsr4 )r. �wol_modeZdict_wol_modes� bit_indexr r r r r$ Y s
zEthtoolWakeOnLan.from_ioctlN)r r r r3 r? r$ r r r r r4 J s r4 ZEthtoolWolMode)r; r< c @ s$ e Zd Zedd� �Zedd� �ZdS )�EthtoolStringBitc
C s� |d }t � }xl|�d�d D ]Z}|d }|�d�}|�d�}x8|d D ],}|d }|�| ||�d�|�d�d �� qHW q W |S )
Nr ZETHTOOL_A_STRSET_STRINGSETSr6 r7 ZETHTOOL_A_STRINGSET_IDZETHTOOL_A_STRINGSET_STRINGSZETHTOOL_A_STRING_INDEXZETHTOOL_A_STRING_VALUE)r r r )r r= �add)r. Znl_string_setsZethtool_strings_set�iZset_idr r r r? j s
z EthtoolStringBit.from_netlinkc s � fdd�t |�D �S )Nc s$ h | ] \}}� |d |d |��qS )� r )r rE �string)r. r r � <setcomp>� s z.EthtoolStringBit.from_ioctl.<locals>.<setcomp>)� enumerate)r. Z string_setsr )r. r r$ � s
zEthtoolStringBit.from_ioctlN)r r r r3 r? r$ r r r r rC g s rC )r r r c s4 e Zd Z� fdd�Zedd� �Zedd� �Z� ZS )�EthtoolLinkInfoc sJ t �|d �}t�|d �}t�|d �}t�|d �}tt| ��| |||||�S )N)r �getr r �superrJ �__new__)r. �port�phyaddr�tp_mdix� tp_mdix_ctrl� transceiver)� __class__r r rM � s
zEthtoolLinkInfo.__new__c C s | |j |j|j|j|jd�S )N)rN rO rP rQ rR )rN Z phy_addressZ eth_tp_mdixZeth_tp_mdix_ctrlrR )r. �
link_settingsr r r r$ � s zEthtoolLinkInfo.from_ioctlc C s8 |d }| |� d�|� d�|� d�|� d�|� d�d�S )Nr ZETHTOOL_A_LINKINFO_PORTZETHTOOL_A_LINKINFO_PHYADDRZETHTOOL_A_LINKINFO_TP_MDIXZ ETHTOOL_A_LINKINFO_TP_MDIX_CTRZ ETHTOOL_A_LINKINFO_TRANSCEIVER)rN rO rP rQ rR )r= )r. � nl_link_moder r r r? � s z EthtoolLinkInfo.from_netlink)r r r rM r3 r$ r? �
__classcell__r r )rS r rJ � s
rJ )rN rO rP rQ rR c s4 e Zd Z� fdd�Zedd� �Zedd� �Z� ZS )�EthtoolLinkModec sF |dks|t ks|tkr d }t�|d �}tt| ��| ||t|�||�S )Nr )�INT32MINUS_UINT32�INT16MINUS_UINT16r rK rL rW rM �bool)r. �speed�duplex�autoneg�supported_ports�supported_modes)rS r r rM � s
zEthtoolLinkMode.__new__c C sv t �|�\}}}t �|�}g }g }x:|D ]2}|jtkrD|�|j� q(|jtkr(|�|j� q(W | |j|j |j
||d�S )N)r[ r\ r] r^ r_ ) r
Zget_link_mode_masksZget_link_mode_bits�typer �appendr r
r[ r\ r] ) r. rT Z
map_supportedZmap_advertisingZmap_lp_advertisingZbits_supportedr^ r_ �bitr r r r$ � s
zEthtoolLinkMode.from_ioctlc C s� |d }g }g }x�|� d�� d�d D ]�}|d }|� d�}|� d�}|� d�}|d k rZq&t| }|j|kr~t�d
|j|� q&|jtkr�|�|j� q&|jtkr&|�|j� q&W | |� d �|� d �|� d
�||d�S )Nr ZETHTOOL_A_LINKMODES_OURSr5 r6 r7 r8 r9 r: Tz0Bit name is not the same as the target: %s <> %sZETHTOOL_A_LINKMODES_SPEEDZETHTOOL_A_LINKMODES_DUPLEXZETHTOOL_A_LINKMODES_AUTONEG)r[ r\ r] r^ r_ ) r= r r �log�errorr` r ra r
) r. rU r^ r_ Z
bitset_bitrB Zbit_nameZ bit_valuerb r r r r? � s:
z EthtoolLinkMode.from_netlink)r r r rM r3 r$ r? rV r r )rS r rW � s
rW )r[ r\ r] r^ r_ c sb e Zd Zddddddddd d
d d d
ddd�Zd� fdd� Zedd� �Zdd� Zedd� �Z� Z S )� EthtoolRingsZETHTOOL_A_RINGS_RX_MAXZETHTOOL_A_RINGS_RX_MINI_MAXZ ETHTOOL_A_RINGS_RX_JUMBO_MAXZETHTOOL_A_RINGS_TX_MAXZETHTOOL_A_RINGS_RXZETHTOOL_A_RINGS_RX_MINIZETHTOOL_A_RINGS_RX_JUMBOZETHTOOL_A_RINGS_TXZETHTOOL_A_RINGS_RX_BUF_LENZ ETHTOOL_A_RINGS_TCP_DATA_SPLITZETHTOOL_A_RINGS_CQE_SIZEZETHTOOL_A_RINGS_TX_PUSHZETHTOOL_A_RINGS_RX_PUSHZ ETHTOOL_A_RINGS_TX_PUSH_BUF_LENZ#ETHTOOL_A_RINGS_TX_PUSH_BUF_LEN_MAX)�rx_max� rx_mini_max� rx_jumbo_max�tx_max�rx�rx_mini�rx_jumbo�tx�
rx_buf_len�tcp_data_split�cqe_size�tx_push�rx_push�tx_push_buf_len�tx_push_buf_len_maxNc s. t t| ��| ||||||||| |
| | |
||�S )N)rL re rM )r. rf rg rh ri rj rk rl rm rn ro rp rq rr rs rt )rS r r rM * s"
zEthtoolRings.__new__c s& � d � | f � fdd�| j �� D ��S )Nr c s i | ]\}}� � |�|�qS r )r= )r �cls_attr� netlink_attr)�nl_ringsr r r! S s z-EthtoolRings.from_netlink.<locals>.<dictcomp>)�nl_attributs_dictr" )r. rw r )rw r r? O s
zEthtoolRings.from_netlinkc C sF t � }x:| j�� D ],\}}t| |�}|d k r|d �||f� qW |S )Nr6 )r rx r" �getattrra )r* Znl_rings_attrsru rv �attrr r r �
to_netlinkY s
zEthtoolRings.to_netlinkc C s t |�}|�d� | f |�S )N�cmd)�dict�pop)r. � ioctl_ringsr r r r$ a s
zEthtoolRings.from_ioctl)NNNNNNNNNNNNNNN)
r r r rx rM r3 r? r{ r$ rV r r )rS r re sB
re )rf rg rh ri rj rk rl rm rn ro rp rq rr rs rt c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
d �Zd!d d
�Zd"dd�Z dd� Z
d#dd�Z d$dd�Z dd� Z
dd� Zdd� Zd d � Zd d � Zd S )%�Ethtoolc C s( t � | _t� | _d| j_| j�� | _d S )N�debug)r
� _with_ioctlr �_with_nlZmodule_err_levelZis_nlethtool_in_kernel�_is_nl_working)r* r r r r+ i s zEthtool.__init__c C s | S )Nr )r* r r r � __enter__o s zEthtool.__enter__c C s | � � d S )N)�close)r* �exc_type� exc_value� tracebackr r r �__exit__r s zEthtool.__exit__c O sF |d kr| j }|dkr t� �y
|||�S tk
r@ t� �Y nX d S )NF)r� r r )r* �f� with_netlink�args�kwargsr r r �_nl_execu s
zEthtool._nl_execNc C sZ y | � | jj||�}t�|�}W n4 tk
rT | j�|� | j�� }t� |�}Y nX |S )N)
r� r� Z get_linkmoderW r? r r� �
change_ifname�get_link_settingsr$ )r* �ifnamer� Z link_moderT r r r �
get_link_mode� s
zEthtool.get_link_modec C sZ y | � | jj||�}t�|�}W n4 tk
rT | j�|� | j�� }t� |�}Y nX |S )N)
r� r� Z get_linkinforJ r? r r� r� r� r$ )r* r� r� Z link_inforT r r r �
get_link_info� s
zEthtool.get_link_infoc C sR y | � | jj||�}t�|�S tk
rL | j�|� | j�� }t�|�S X d S )N) r� r� Z
get_stringsetrC r? r r� r� r$ )r* r� r� Z
stringsetsr r r �get_strings_set� s
zEthtool.get_strings_setc C sj | j }|dkr>y| j�|�}t�|�S tk
r< d}Y nX |dkrf| j�|� | j�� }t�|�S d S )NTF) r� r� �get_wolr4 r? r r� r� r$ )r* r� Z
nl_workingZwolrA r r r r� � s
zEthtool.get_wolc C sZ y | � | jj||�}t�|�}W n4 tk
rT | j�|� | j�� }t�|�}Y nX |S )N) r� r� � get_ringsre r? r r� r� r$ )r* r� r� �ringsZ
rings_infor r r r� � s
zEthtool.get_ringsc K s� y&t f |��� }| �| jj|||� W nl tk
r� | j�|� | j�� }x4|� � D ](\}}||�
� krV|| |krV|||< qVW | j�|� Y nX d S )N) re r{ r� r� � set_ringsr r� r� r� r" �keys)r* r� r� r� r� r r r r r r r� � s
zEthtool.set_ringsc C s | j �|� t�| j �� �S )N)r� r� r- r$ � get_features)r* r� r r r r� � s zEthtool.get_featuresc C s2 | j �|� | j �� }t�||� | j �|� d S )N)r� r� r� r- r& � set_features)r* r� r/ r1 r r r r� � s
zEthtool.set_featuresc C s | j �|� t�| j �� �S )N)r� r� r r$ � get_coalesce)r* r� r r r r� � s zEthtool.get_coalescec C s2 | j �|� | j �� }t�||� | j �|� d S )N)r� r� r� r r& � set_coalesce)r* r� r% r# r r r r� � s
zEthtool.set_coalescec C s | j �� | j�� d S )N)r� r� r� )r* r r r r� � s
z
Ethtool.close)N)N)N)N)N)r r r r+ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� h s
r� ))�logging� collectionsr �ctypesr r Zpyroute2.ethtool.commonr r r r r r
r Zpyroute2.ethtool.ioctlr r
Zpyroute2.netlink.exceptionsr Z pyroute2.netlink.generic.ethtoolr r r rX rY � getLoggerr rc r � Exceptionr �objectr r( r- r4 rC rJ rW re r� r r r r �<module> s> $
'
M
b