File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/__pycache__/kmod.cpython-310.pyc
o
�N�gn � @ s� d Z ddlZddlZddlZddlZddlZddlmZ e� e
�Z dd� Z dd� Z
dd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd$dd �Zd$d d �Zd d � Zd%d"d#�ZdS )&z'
Module to manage Linux kernel modules
� N)�CommandExecutionErrorc C s t d dkS )z$
Only runs on Linux systems
Zkernel�Linux��
__grains__� r r �E/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/kmod.py� __virtual__ s r c C sD t � }t � }| D ] }|�|d � q|D ] }|�|d � q|| S �zy
Return a list of the new modules, pass an lsmod dict before running
modprobe and one after modprobe has run
�module��set�add��pre_mods� post_modsZpreZpost�modr r r � _new_mods � r c C sD t � }t � }| D ] }|�|d � q|D ] }|�|d � q|| S r r r r r r �_rm_mods% r r c C s dt v rdS dS )zK
Return location of modules config file.
Default: /etc/modules
Zsystemdz%/etc/modules-load.d/salt_managed.confz /etc/modulesr r r r r �_get_modules_conf3 s r c C s | � � dkrdS | �� d S )z�
Return module name and strip configuration. It is possible insert modules
in this format:
bonding mode=4 miimon=1000
This method return only 'bonding'
� Fr )�strip�split�r r r r �_strip_module_name= s r c C s� t � }tj�|�std |� t| �}|r |td�v s |t� vr#t� S t �
| �}td |d|� d�dd�r?td ||� |hS td || � |hS ) zj
Add module to configuration file to make it persistent. If module is
commented uncomment it.
z
file.touchTz file.searchz^#[ ]*z[ ]*$)Z multilinezfile.uncommentz file.append) r �os�path�exists�__salt__r �mod_list� availabler �re�escape)r �conf�mod_name�
escape_modr r r �_set_persistent_moduleI s
�r& c C sl t � }t| �}|r|td�vrt� S t�| �}|r'td |d|� d�� |hS td |d|� d�d� |hS )zj
Remove module from configuration file. If comment is true only comment line
where module is.
Tz file.commentz^[ ]*z[ ]?zfile.sedr )r r r r r! r" r )r �commentr# r$ r% r r r �_remove_persistent_module] s
�r( c C s& t jj�| �}|std| � d���|S )zO
Utility function wrapper to error out early if a command is not found
z Command 'z' cannot be found)�salt�utilsr �whichr )�cmdZ_cmdr r r �_whichn s r- c
C sT g } t j�dt �� d �}t j�|d�}t j�|�rDtjj�|d��}|D ]}| � t j�
|�dd� � q%W d � n1 s?w Y tjj� |�D ] \}}}|D ]}d|v rh| � |d|� d�� �
dd �� qRqKd
td v r�d t �� d d
d� d } tjj� | �D ] \}}}|D ]}d|v r�| � |d|� d�� �
dd �� q�q�tt| ��S )z�
Return a list of all available kernel modules
CLI Example:
.. code-block:: bash
salt '*' kmod.available
z
/lib/modules/� zmodules.builtin�rN���z.ko�-�_ZArchZ os_familyz/lib/modules/extramodules-r � z-ARCH)r r �join�unamer r) r* �files�fopen�append�basenameZos_walk�index�replacer �sorted�list)
�retZmod_dirZ
built_in_file�f�line�root�dirsr6 Zfn_Z mod_dir_archr r r r x s0
� � �� �� r c C �
| t � v S )z�
Check to see if the specified kernel module is available
CLI Example:
.. code-block:: bash
salt '*' kmod.check_available kvm
)r r r r r �check_available� �
rD c C s� g } t d td���� D ]9}|�� }t|�dksq |d dkr q |d |d |d d�}t|�dkr<|d �d �|d
< ng |d
< | �|� q | S ) z�
Return a dict containing information about currently loaded modules
CLI Example:
.. code-block:: bash
salt '*' kmod.lsmod
zcmd.run�lsmodr. r ZModule� )�sizer
Zdepcountr3 �,Zdeps)r r- �
splitlinesr �lenr8 )r> r@ �compsZmdatr r r rF � s
� rF Fc C s� t � }| rRt� }tj�|�rQz3tjj�|d��!}|D ]}|� � }t
|�}|� d�s0|r0|� |� qW d � n1 s;w Y W n t
yP t�d|� Y nw n
t� D ] }|� |d � qUtt|��S )z�
Return a list of the loaded module names
only_persist
Only return the list of loaded persistent modules
CLI Example:
.. code-block:: bash
salt '*' kmod.mod_list
r/ �#Nz-kmod module could not open modules file at %sr
)r r r r r r) r* r6 r7 r r �
startswithr
�OSError�log�errorrF r< r= )Z only_persist�modsr# Z modules_filer@ r$ r r r r r � s,
�� �� ��
r c C sp t � }td d�td�| �dd�}|d dkr0t � }t||�}t� }|r(t| �}tt||B ��S d�| |d �S )
z�
Load the specified kernel module
mod
Name of module to add
persist
Write module to /etc/modules to make it load on system reboot
CLI Example:
.. code-block:: bash
salt '*' kmod.load kvm
� cmd.run_all�{} {}ZmodprobeF�Z python_shell�retcoder zError loading module {}: {}�stderr) rF r �formatr- r r r& r< r= )r �persistr �resr rR � persist_modsr r r �load� s �
r\ c C rC )z�
Check to see if the specified kernel module is loaded
CLI Example:
.. code-block:: bash
salt '*' kmod.is_loaded kvm
)r r r r r � is_loaded rE r] Tc C sr t � }td d�td�| �dd�}|d dkr1t � }t||�}t� }|r)t| |�}tt||B ��S d�| |d �S )
aE
Remove the specified kernel module
mod
Name of module to remove
persist
Also remove module from /etc/modules
comment
If persist is set don't remove line from /etc/modules but only
comment it
CLI Example:
.. code-block:: bash
salt '*' kmod.remove kvm
rS rT ZrmmodFrU rV r z Error removing module {}: {}rW ) rF r rX r- r r r( r< r= )r rY r' r rZ r rR r[ r r r �remove s �
r^ )F)FT)�__doc__�loggingr r! Zsalt.utils.filesr) Zsalt.utils.pathZsalt.exceptionsr � getLogger�__name__rP r r r r r r& r( r- r rD rF r r\ r] r^ r r r r �<module> s.
%