| Current File : //lib/python3.9/site-packages/firewall/core/__pycache__/modules.cpython-39.pyc |
a
���g� � @ sB d Z dgZddlmZ ddlmZ ddlmZ G dd� de�Z dS )zmodules backend�modules� )�runProg)�log)�COMMANDSc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )r c C s t d | _t d | _d S )NZmodprobeZrmmod)r �
_load_command�_unload_command��self� r
�9/usr/lib/python3.9/site-packages/firewall/core/modules.py�__init__ s
zmodules.__init__c C s
d| j S )Nz%s)� __class__r r
r
r �__repr__$ s zmodules.__repr__c C s� g }i }z�t dd��x}|D ]b}|s& q~|�� }|�� }|�|d � |d dkrp|d �d�dd� ||d < qg ||d < qW d � n1 s�0 Y W n ty� Y n0 ||fS ) z6 get all loaded kernel modules and their dependencies z
/proc/modules�rr � �-�,N���)�open�strip�split�append�FileNotFoundError)r �mods�deps�f�lineZsplitsr
r
r �loaded_modules' s 0 zmodules.loaded_modulesc C s" t �d| j| j|� t| j|g�S �Nz %s: %s %s)r �debug2r
r r �r �moduler
r
r � load_module< s zmodules.load_modulec C s" t �d| j| j|� t| j|g�S r )r r r
r r r r
r
r �
unload_module@ s zmodules.unload_modulec C sP ||vr dS || D ]$}| � |||� ||vr|�|� q||vrL|�|� dS )z get all dependants of a module N)�get_depsr )r r! r �ret�modr
r
r r$ D s zmodules.get_depsc C s� g }| � � \}}| �d||� dD ]"}||v r"|�|� |�d|� q"|D ]V}|dv s�|�d�s�|�d�s�|�d�s�|�d�s�|�d �s�|�d
�rJ| �|||� qJ|S ) z) get all loaded firewall-related modules Z nf_conntrack)Znf_conntrack_ipv4Znf_conntrack_ipv6r )Z ip_tablesZ
ip6_tablesZebtablesZiptable_Z ip6table_Znf_Zxt_Zipt_Zip6t_)r r$ �remove�insert�
startswith)r r Zmods2r Zbad_bad_moduler& r
r
r �get_firewall_modulesO s.
������z modules.get_firewall_modulesc C s: | � � D ],}| �|�\}}|dkrt�d||f � qdS )z% unload all firewall-related modules r z Failed to unload module '%s': %sN)r* r# r Zdebug1)r r! �statusr% r
r
r �unload_firewall_modulesd s z modules.unload_firewall_modulesN) �__name__�
__module__� __qualname__r r r r" r# r$ r* r, r
r
r
r r s N)
�__doc__�__all__Zfirewall.core.progr Zfirewall.core.loggerr Zfirewall.configr �objectr r
r
r
r �<module> s