3 Y�j��@sBdZdgZddlmZddlmZddlmZGdd�de�Z dS)zmodules backend�modules�)�runProg)�log)�COMMANDSc@sLeZdZdd�Zdd�Zdd�Zdd�Zd d �Zdd�Zd d�Z dd�Z dS)rcCstd|_td|_dS)NZmodprobeZrmmod)r� _load_command�_unload_command)�self�r �/usr/lib/python3.6/modules.py�__init__s zmodules.__init__cCs d|jS)Nz%s)� __class__)rr r r �__repr__$szmodules.__repr__cCs�g}i}y�tdd��p}xh|D]`}|s&P|j�}|j�}|j|d�|ddkrp|djd�dd �||d<qg||d<qWWdQRXWntk r�YnX||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 zmodules.loaded_modulescCs"tjd|j|j|�t|j|g�S)Nz %s: %s %s)r�debug2rrr)r�moduler r r �load_module<szmodules.load_modulecCs"tjd|j|j|�t|j|g�S)Nz %s: %s %s)rrrrr)rrr r r � unload_module@szmodules.unload_modulecCsT||krdSx0||D]$}|j|||�||kr|j|�qW||krP|j|�dS)z get all dependants of a module N)�get_depsr)rrr�ret�modr r r r"Dszmodules.get_depscCs�g}|j�\}}|jd||�x*dD]"}||kr$|j|�|jd|�q$Wx^|D]V}|dks�|jd�s�|jd �s�|jd �s�|jd�s�|jd�s�|jd �rP|j|||�qPW|S)z) get all loaded firewall-related modules Znf_conntrack�nf_conntrack_ipv4�nf_conntrack_ipv6r� ip_tables� ip6_tables�ebtablesZiptable_Z ip6table_Znf_Zxt_Zipt_Zip6t_)r%r&r)r'r(r))rr"�remove�insert� startswith)rrZmods2rZbad_bad_moduler$r r r �get_firewall_modulesOs zmodules.get_firewall_modulescCs>x8|j�D],}|j|�\}}|dkr tjd||f�q WdS)z% unload all firewall-related modules rz Failed to unload module '%s': %sN)r-r!rZdebug1)rrZstatusr#r r r �unload_firewall_modulesdszmodules.unload_firewall_modulesN)�__name__� __module__�__qualname__rr rr r!r"r-r.r r r r rsN) �__doc__�__all__Zfirewall.core.progrZfirewall.core.loggerrZfirewall.configr�objectrr r r r �s