File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/__pycache__/ethtool.cpython-310.pyc
o
�N�g^, � @ s� d Z ddlZddlZddlZddlmZ zddlZdZW n e y' dZY nw e�
e �Z i dd�dd�dd �d
d �d d
�dd�dd�dd�dd�dd�dd�dd�d d �d d �d d!�d"d#�d$d%�d&d'd(d)d*d+��Z
i Ze
�� D ]\ZZeee< qpd,d-d.d/d0d1d2d3d4�Zi Ze�� D ]\ZZeee< q�d5Zd6d7� Zd8d9� Zd:d;� Zd<d=� Zd>d?� Zd@dA� ZdBdC� ZdDdE� Z dFdG� Z dHdI� Z dJdK� Z dLdM� Z dNdO� Z!dPdQ� Z"dS )Rz�
Module for running ethtool command
.. versionadded:: 2016.3.0
:codeauthor: Krzysztof Pawlowski <msciciel@msciciel.eu>
:maturity: new
:depends: python-ethtool
:platform: linux
� N)�CommandExecutionErrorTFZ
pkt_rate_highZ pkt_rate_lowZsample_intervalZrate_sample_intervalZrx_usecsZrx_coalesce_usecsZ
rx_usecs_highZrx_coalesce_usecs_highZ rx_usecs_irqZrx_coalesce_usecs_irqZ rx_usecs_lowZrx_coalesce_usecs_lowZ rx_framesZrx_max_coalesced_framesZrx_frames_highZ rx_max_coalesced_frames_highZ
rx_frames_irgZrx_max_coalesced_frames_irqZ
rx_frames_lowZrx_max_coalesced_frames_lowZstats_block_usecsZstats_block_coalesce_usecsZtx_usecsZtx_coalesce_usecsZ
tx_usecs_highZtx_coalesce_usecs_highZ tx_usecs_irqZtx_coalesce_usecs_irqZ tx_usecs_lowZtx_coalesce_usecs_lowZ tx_framesZtx_max_coalesced_framesZ tx_max_coalesced_frames_highZtx_max_coalesced_frames_irqZtx_max_coalesced_frames_lowZuse_adaptive_rx_coalesceZuse_adaptive_tx_coalesce)Ztx_frames_highZ
tx_frames_irqZ
tx_frames_lowZ adaptive_rxZ adaptive_txZ
rx_pendingZrx_max_pendingZrx_mini_pendingZrx_mini_max_pendingZrx_jumbo_pendingZrx_jumbo_max_pendingZ
tx_pendingZtx_max_pending)�rxZrx_maxZrx_miniZ rx_mini_maxZrx_jumboZ rx_jumbo_max�txZtx_max�ethtoolc C s t rtS dS )z>
Only load this module if python-ethtool is installed
)FzKThe ethtool module could not be loaded: ethtool python libraries not found.)� HAS_ETHTOOL�__virtualname__� r r �H/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/ethtool.py� __virtual__K s r
c C �Z zt �| �}W n ty t�d| � Y dS w i }|�� D ] \}}|| |t| < q |S )z�
Queries the specified network device for rx/tx ring parameter information
CLI Example:
.. code-block:: bash
salt '*' ethtool.show_ring <devname>
�#Ring parameters not supported on %s�
Not supported)r �
get_ringparam�OSError�log�error�items�ethtool_ring_remap)�devname�ring�ret�key�valuer r r � show_ringY � �r c C r )z�
Queries the specified network device for coalescing information
CLI Example:
.. code-block:: bash
salt '*' ethtool.show_coalesce <devname>
�(Interrupt coalescing not supported on %sr
)r � get_coalescer r r r �ethtool_coalesce_remap)r �coalescer r r r r r �
show_coalesceq r r c C sn zt �| �}W n ty t�d| � Y dS w zt �| �}W n ty/ t�d| � Y dS w ||d�}|S )z�
Queries the specified network device for associated driver information
CLI Example:
.. code-block:: bash
salt '*' ethtool.show_driver <devname>
z(Driver information not implemented on %szNot implementedz"Bus information no available on %sz
Not available)ZdriverZbus_info)r Z
get_moduler r r Z get_businfo)r �moduleZbusinfor r r r � show_driver� s � ��r! c K � zt �| �}W n ty t�d| � Y dS w d}|�� D ] \}}|tv r:t| }||v r:|| |kr:|||< d}q z
|rDt �| |� t| �W S tyY t�d| |� Y dS w )z�
Changes the rx/tx ring parameters of the specified network device
CLI Example:
.. code-block:: bash
salt '*' ethtool.set_ring <devname> [rx=N] [rx_mini=N] [rx_jumbo=N] [tx=N]
r r
FTz Invalid ring arguments on %s: %s�Invalid arguments) r r r r r r �ethtool_ring_mapZ
set_ringparamr )r �kwargsr �changed�paramr r r r �set_ring� s. � �
�r( c K r" )aY
Changes the coalescing settings of the specified network device
CLI Example:
.. code-block:: bash
salt '*' ethtool.set_coalesce <devname> [adaptive_rx=on|off] [adaptive_tx=on|off] [rx_usecs=N] [rx_frames=N]
[rx_usecs_irq=N] [rx_frames_irq=N] [tx_usecs=N] [tx_frames=N] [tx_usecs_irq=N] [tx_frames_irq=N]
[stats_block_usecs=N] [pkt_rate_low=N] [rx_usecs_low=N] [rx_frames_low=N] [tx_usecs_low=N] [tx_frames_low=N]
[pkt_rate_high=N] [rx_usecs_high=N] [rx_frames_high=N] [tx_usecs_high=N] [tx_frames_high=N]
[sample_interval=N]
r r
FTz$Invalid coalesce arguments on %s: %sr# ) r r r r r r �ethtool_coalesce_map� set_coalescer )r r% r r& r' r r r r r* � s. � �
�r* c C s� z t �| �rdp d}W n ty d}Y nw z t �| �r dp d}W n ty- d}Y nw z t �| �r6dp7d}W n tyD d}Y nw z t �| �rMdpNd}W n ty[ d}Y nw ||||d�}|S )z�
Queries the specified network device for the state of protocol offload and other features
CLI Example:
.. code-block:: bash
salt '*' ethtool.show_offload <devname>
�on�offz
not supported)Zscatter_gather�tcp_segmentation_offloadZudp_fragmentation_offloadZ generic_segmentation_offload)r Zget_sgr Zget_tsoZget_ufoZget_gso)r �sg�tso�ufo�gsoZoffloadr r r � show_offload� s4 � � � ��r2 c K sZ |� � D ]$\}}|dkr(|dkrdpd}zt�| |� W q ty' Y dS w qt| �S )z�
Changes the offload parameters and other features of the specified network device
CLI Example:
.. code-block:: bash
salt '*' ethtool.set_offload <devname> tcp_segmentation_offload=on
r- r+ � r r
)r r Zset_tsor r2 )r r% r' r r r r � set_offload s ��r4 c O s� t jj�d�}|s
td��d�dd� |D ��}d�dd� |�� D ��}|� d|� d| � d|� ��� }td |dd �� � }|rK|d
�
d �rKt|d
��|S ) z5
Helper function to build an ethtool command
r z!Command 'ethtool' cannot be found� c s s � | ]}|V qd S �Nr )�.0�argr r r � <genexpr>: s � z#_ethtool_command.<locals>.<genexpr>c s s" � | ] \}}|� d |� �V qdS )r5 Nr )r7 r �valr r r r9 ; s � zcmd.runT)Zignore_retcoder ZCannot) �salt�utils�path�whichr �joinr �stripZ__salt__�
splitlines�
startswith)r �argsr% r Zswitches�params�cmdr r r r �_ethtool_command3 s rF c C s^ i }|� � D ] \}}|�� }|| v r#|du rd}n|du r d}|||< q|s-td| � ���|S )z�
Helper function to validate parameters to ethtool commands. Boolean values
will be transformed into ``on`` and ``off`` to match expected syntax.
Tr+ Fr, z,None of the valid parameters were provided: )r �lowerr )� valid_paramsr% Z validatedr r: r r r �_validate_paramsC s ��rI c C sR i }t | d�}|dd� D ]}|�� r&dd� |�dd�D �\}}|dk||< q
|S )z�
.. versionadded:: 3006.0
Queries the specified network device for associated pause information
CLI Example:
.. code-block:: bash
salt '*' ethtool.show_pause <devname>
z-ar3 Nc s � � | ]}|� � V qd S r6 �r@ �r7 �sr r r r9 j � � z show_pause.<locals>.<genexpr>�:r+ �rF r@ �split)r �data�content�liner r r r r �
show_pauseX s
�rU c K s0 g d�}t ||�}t| dfi |��}|sdS |S )z�
.. versionadded:: 3006.0
Changes the pause parameters of the specified network device
CLI Example:
.. code-block:: bash
salt '*' ethtool.set_pause <devname> autoneg=off rx=off tx=off
)Zautonegr r z-AT)rI rF �r r% rH rD r r r r � set_pausep s
rW c C s| i }t | d�}|dd� D ].}d|v r;dd� |�� �dd�D �\}}d|v }|r0|�� d �� }|d k|d
�||�� < q
|S ) z�
.. versionadded:: 3006.0
Queries the specified network device for associated feature information
CLI Example:
.. code-block:: bash
salt '*' ethtool.show_features <devname>
z-kr3 NrO c s rJ r6 rK rL r r r r9 � rN z show_features.<locals>.<genexpr>�fixedr r+ )r+ rX rP )r rR rS rT r r rX r r r �
show_features� s
�rY c K s8 g d�}t ||�}t| dfi |��}|sdS tj�|�S )z�
.. versionadded:: 3006.0
Changes the feature parameters of the specified network device
CLI Example:
.. code-block:: bash
salt '*' ethtool.set_feature <devname> sg=off
) r r r. r/ r0 r1 ZgroZlroZrxvlanZtxvlanZntupleZrxhashz-KT)rI rF �os�linesepr? rV r r r � set_feature� s
r\ )#�__doc__�loggingrZ Zsalt.utils.pathr; Zsalt.exceptionsr r r � ImportError� getLogger�__name__r r) r r �k�vr$ r r r
r r r! r( r* r2 r4 rF rI rU rW rY r\ r r r r �<module> s� �
�������� �
� � �
������
�
#))