Edit File: base.cpython-311.pyc
� �gh � � � d dl mZmZ d dlmZ d dlmZmZ G d� de� � Z G d� de � � Z G d� d e � � Z G d � d e � � Z G d � d e�� � Z dS )� )�ABCMeta�abstractmethod)� RulesLock)�IP� IPVersionc �( � � e Zd ZdZdef� fd� Z� xZS )� FirewallErrorz(Root exception class for firewall errors� ip_versionc �H �� t � � j |i |�� || _ d S �N)�super�__init__r )�selfr �args�kwargs� __class__s ��W/opt/imunify360/venv/lib/python3.11/site-packages/im360/internals/core/firewall/base.pyr zFirewallError.__init__ s* �� ������$�)�&�)�)�)�$����� )�__name__� __module__� __qualname__�__doc__r r � __classcell__�r s @r r r sJ �� � � � � �2�2�%�9� %� %� %� %� %� %� %� %� %� %r r c � � e Zd ZdZdS )�FirewallTemporaryErrorz5Root exception class for temporary (transient) errorsN�r r r r � r r r r s � � � � � �?�?��Dr r c � � e Zd ZdZdS )� FirewallCommandNotFoundErrorzCRaised if underlying command execution fails with FileNotFoundErrorNr r r r r r s � � � � � �M�M��Dr r c �, � � e Zd ZdZdedef� fd� Z� xZS )�FirewallBatchCommandErrorz%Raised when a batch of command fails.r �commandc �R �� t � � j ||g|�R i |�� || _ d S r )r r r# )r r r# r r r s �r r z"FirewallBatchCommandError.__init__ s7 �� �������W�>�t�>�>�>�v�>�>�>� �� � � r )r r r r r �strr r r s @r r" r" sQ �� � � � � �/�/� �9� �s� � � � � � � � � � r r" c � � e Zd ZdZ e� � Zej fd�Ze d� � � Z e d� � � Z e d� � � Z e d� � � Z e d� � � Ze d� � � Ze d � � � Ze d � � � Ze d � � � Zd � Zd � ZdS )�AbstractFirewallza Abstract class that defines required interface for iptables ruleset editing classes c � � || _ d S r )r )r r s r r zAbstractFirewall.__init__+ s � �$����r c � � K � d S r r �r r r s r �has_rulezAbstractFirewall.has_rule. � � � � � �r c � � K � d S r r r* s r � has_chainzAbstractFirewall.has_chain2 r, r c � � K � d S r r r* s r � append_rulez AbstractFirewall.append_rule6 r, r c � � K � d S r r r* s r � insert_rulez AbstractFirewall.insert_rule: r, r c � � K � d S r r r* s r � delete_rulez AbstractFirewall.delete_rule> r, r c � � K � d S r r r* s r � flush_chainz AbstractFirewall.flush_chainB r, r c � � K � d S r r r* s r � create_chainz AbstractFirewall.create_chainF r, r c � � K � d S r r r* s r � delete_chainz AbstractFirewall.delete_chainJ r, r c � � K � d S r r )r �recordss r �commitzAbstractFirewall.commitN r, r c � �H K � | j � � � � d {V �� | S r )�_lock�acquire)r s r � __aenter__zAbstractFirewall.__aenter__R s1 � � � ��j� � �"�"�"�"�"�"�"�"�"�� r c � �< K � | j � � � d S r )r? �release)r �exc_type�exc_val�exc_tbs r � __aexit__zAbstractFirewall.__aexit__V s � � � � � ��� � � � � r N)r r r r r r? r �V4r r r+ r. r0 r2 r4 r6 r8 r: r= rA rG r r r r' r' # sU � � � � � �� � �I�K�K�E�"$�%� %� %� %� %� � � � �^� � � � � �^� � � � � �^� � � � � �^� � � � � �^� � � � � �^� � � � � �^� � � � � �^� � � � � �^� �� � � � � � � r r' )� metaclassN)�abcr r � im360.utilsr � defence360agent.utils.validater r � Exceptionr r r r" r' r r r �<module>rN s( �� '� '� '� '� '� '� '� '� !� !� !� !� !� !� 8� 8� 8� 8� 8� 8� 8� 8�%� %� %� %� %�I� %� %� %� � � � � �]� � � � � � � � �=� � � � � � � � � � � � �4 � 4 � 4 � 4 � 4 �� 4 � 4 � 4 � 4 � 4 � 4 r