Edit File: port_ips_deny_mode.cpython-311.pyc
� �gh& � �� � d dl Z d dlZd dlmZ d dlmZmZmZ ej e � � Z dxZ \ Z Z Z G d� de� � Zed Zeeeeee f f Z G d� d� � ZdS ) � N)�Path)�List�Literal�Union��tcp�udp�allc � � e Zd ZdS )�IpPortDenyModeErrorN)�__name__� __module__� __qualname__� � �S/opt/imunify360/venv/lib/python3.11/site-packages/im360/model/port_ips_deny_mode.pyr r s � � � � � ��Dr r c � � e Zd ZdZedefd�� � Zedededefd�� � Zede e d e e d f de fd �� � Z d S ) �WhitelistPortIPsDenyModez /etc/imunify360/whitelist/ports�returnc �� � t i t i i}t | j � � � d� � D ]*}| � |� � � |� � }�+|S )z Read port:networks from the filez*.txt)�TCP�UDPr � _LIST_PATH�glob� _load_file�absolute)�cls�port_to_networks�paths r �loadz WhitelistPortIPsDenyMode.load sg � � ��S�"�-���� �(�(�-�-�g�6�6� � �D� "�~�~�� � ���!1� � � � � � r r r c � � t |� � � � � � � � D �]�\ }}|� d� � d � � � }|s�6g }d}|� d� � \ }} } t t g} | � d� � \ } } }| � � � � � � } | t v r| t k r| g} n| }| sd} t |� � }|� d� � D ]U}|� � � }|r= t j |� � |� |� � �C# t $ r d}Y �Qw xY w�V| D ]g} |rc|| � |g � � }|� |� � t% t' t) |� � � � � � || |< �hn# t $ r d}Y nw xY w|s t* � d||� � ���|S )N�#r T�:F�,z1Wrong IP/Subnet format in %s:%s. Use CIDR format.)� enumerate�open� readlines� partition�stripr r �lower�PROTOS�ALL�int�split� ipaddress� ip_network�append� ValueError� setdefault�extend�sorted�list�set�logger�error)r r r �line_num�line�item� network_list�valid�port� semicolon�tail�protos�proto�_�networks�network�existing_portss r r z#WhitelistPortIPsDenyMode._load_file! sY � � (�� � � � �(=�(=�(?�(?� @� @� 3 � 3 �N�H�d��>�>�#�&�&�q�)�/�/�1�1�D�� �� �L��E�$(�N�N�3�$7�$7� !�D�)�T�#&��*�F�!%� � ��!4�!4� �E�1�h� �;�;�=�=�.�.�0�0�E������C�<�<� #�W�F�� �� � � �� ��4�y�y�� (�~�~�c� 2� 2� *� *�G� %�m�m�o�o�G� � *�*� %� 0�� 9� 9� 9� (� /� /�� 8� 8� 8� 8�� )� *� *� *�$)�E�E�E�*���� *� $� � �E�#� �)9�%�)@�)K�)K� �"�*� *�� '�-�-�l�;�;�;�8>� ��^�!4�!4� 5� 5�9� 9�(��/��5�� �� � � � � ���� ����, � �� � �G�� �� � �� � s$ �3G�0)E� E)�(E)� G'�&G'rC r �r r c �. �� |x t |t � � r t d� |D � � � � r |}n1xxdk rn xdk rn n |g}ndk rddg}n t d� � �| � � � �t �fd�|D � � � � S )Nc 3 � K � | ]}|d v V � � dS )rH Nr )�.0�ps r � <genexpr>z1WhitelistPortIPsDenyMode.count.<locals>.<genexpr>d s8 � � � � 6� 6�()��^�#�6� 6� 6� 6� 6� 6r r r r z Invalid inputc 3 �z �K � | ]5}t d � �| � � � D � � � � V � �6dS )c 3 �4 K � | ]}t |� � V � �d S )N)�len)rK rE s r rM z;WhitelistPortIPsDenyMode.count.<locals>.<genexpr>.<genexpr>p s( � � � �I�I�(��H� � �I�I�I�I�I�Ir N)�sum�values)rK rL �proto_to_portss �r rM z1WhitelistPortIPsDenyMode.count.<locals>.<genexpr>o s_ �� � � � � �� �I�I�n�Q�.?�.F�.F�.H�.H�I�I�I� I� I� � � � � � r )� isinstancer6 r r2 r rQ )r rC rB rS s @r �countz WhitelistPortIPsDenyMode.count\ s� �� � ���j���-�-��#� 6� 6�-2�6� 6� 6� 3� 3��� ��� ��������������� ������� �� ���� � �1�1�1� ������� � � � �� � � � � � r N)r r r r � classmethod�PortNetworkMapr r r r r r r- rU r r r r r s� � � � � � �2�J�� �^� � � � �[� � �8 ��8 �+9�8 � �8 � 8 � 8 � �[�8 �t � �� �'� (�$�w�|�/D�*E� E� � � � � � � �[� � � r r )r/ �logging�pathlibr �typingr r r � getLoggerr r8 r+ r r r, � Exceptionr �Proto�dictr- �strrW r r r r �<module>r` s� �� � � � � ���� � � � � � � '� '� '� '� '� '� '� '� '� '� �� �8� $� $�� 1� 0�� �#�s�C� � � � � �)� � � � �#�$���e�T�#�t�C�y�.� 1�1�2��^ � ^ � ^ � ^ � ^ � ^ � ^ � ^ � ^ � ^ r