a
i @ sB d Z dgZddlmZ ddlmZ ddlmZ G dd deZ dS )zmodules backendmodules )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_commandself 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 }zt ddx}|D ]b}|s& q~| }| }||d |d dkrp|d ddd ||d < qg ||d < qW d n1 s0 Y W n ty Y n0 ||fS ) z6 get all loaded kernel modules and their dependencies z
/proc/modulesrr -,N)openstripsplitappendFileNotFoundError)r modsdepsflineZsplitsr
r
r loaded_modules' s 0 zmodules.loaded_modulesc C s" t d| j| j| t| j|gS 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|gS 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 retmodr
r
r r$ D s zmodules.get_depsc C s g }| \}}| d|| dD ]"}||v r"|| |d| q"|D ]V}|dv s|ds|ds|ds|ds|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$ removeinsert
startswith)r r Zmods2r Zbad_bad_moduler& r
r
r get_firewall_modulesO s.
z modules.get_firewall_modulesc C s: | D ],}| |\}}|dkrtd||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 s