Edit File: modules.cpython-36.pyc
3 Y e� � @ 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 � /usr/lib/python3.6/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 }y�t dd��p}xh|D ]`}|s&P |j� }|j� }|j|d � |d dkrp|d jd�dd � ||d < q g ||d < q W W dQ R X W n tk r� Y nX ||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_modulesc C s" t jd| 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 jd| j| j|� t| j|g�S )Nz %s: %s %s)r r r r r )r r r r r � unload_module@ s zmodules.unload_modulec C sT ||kr dS x0|| D ]$}| j |||� ||kr|j|� qW ||krP|j|� 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 }| j � \}}| jd||� x*dD ]"}||kr$|j|� |jd|� q$W x^|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 Z nf_conntrack�nf_conntrack_ipv4�nf_conntrack_ipv6r � ip_tables� ip6_tables�ebtablesZiptable_Z ip6table_Znf_Zxt_Zipt_Zip6t_)r% r& r )r' r( r) )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> x8| j � D ],}| j|�\}}|dkr tjd||f � q W dS )z% unload all firewall-related modules r z Failed to unload module '%s': %sN)r- r! r Zdebug1)r r Zstatusr# 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