File: //proc/self/root/lib/python3/dist-packages/firewall/core/__pycache__/modules.cpython-310.pyc
o
bhAb� � @ 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 )N�modprobe�rmmod)r �
_load_command�_unload_command��self� r �7/usr/lib/python3/dist-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 }t dd��I}|D ]1}|s n6|�� }|�� }|�|d � |d dkr7|d �d�dd� ||d < q g ||d < q W d � ||fS W d � ||fS 1 sUw Y ||fS ) z6 get all loaded kernel modules and their dependencies z
/proc/modules�rr � �-�,N���)�open�strip�split�append)r �mods�deps�f�line�splitsr r r
�loaded_modules' s( �
� �
� � zmodules.loaded_modulesc C �" t �d| j| j|� t| j|g�S �Nz %s: %s %s)r �debug2r r r �r �moduler r r
� load_module9 � zmodules.load_modulec C r r! )r r" r r r r# r r r
�
unload_module= r& zmodules.unload_modulec C sT ||vrdS || D ]}| � |||� ||vr |�|� q
||vr(|�|� dS dS )z get all dependants of a module N)�get_depsr )r r$ r �ret�modr r r
r( A s
��zmodules.get_depsc C s� g }| � � \}}| �d||� dD ]}||v r"|�|� |�d|� q|D ]+}|dv sI|�d�sI|�d�sI|�d�sI|�d�sI|�d �sI|�d
�rP| �|||� q%|S ) z) get all loaded firewall-related modules � nf_conntrack)�nf_conntrack_ipv4�nf_conntrack_ipv6r )� ip_tables�
ip6_tables�ebtables�iptable_� ip6table_�nf_�xt_�ipt_�ip6t_)r r( �remove�insert�
startswith)r r �mods2r �bad_bad_moduler* r r r
�get_firewall_modulesL s2
��������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 �debug1)r r$ �statusr) r r r
�unload_firewall_modulesa 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__�firewall.core.progr �firewall.core.loggerr �firewall.configr �objectr r r r r
�<module> s