| Current File : //opt/imunify360/venv/lib/python3.11/site-packages/im360/utils/__pycache__/net.cpython-311.pyc |
�
�*(i! � � � d dl Z d dlZd dlZd dlmZ d dlmZmZmZm Z m
Z
m Z m Z d dl
mZmZmZmZ d dlZd dlmZ d dlmZmZ e j e� � Z ej dej � � Z d\ Z Z Z d \ Z!Z"ee e f Z#d
Z$d d ee fd �Z%d
eeee
f fd�Z&d� Z' ed�� � d� � � Z(de)fd�Z*d� Z+deee
f fd�Z,dee e f d
ee)e)e)f fd�Z-de)de)d e)d
e#fd �Z.de/d
ede e f fd �Z0dS ) � N)� lru_cache)�
IPV4LENGTH�
IPV6LENGTH� IPv4Address� IPv4Network� IPv6Address� IPv6Network�
ip_network)�Iterable�Optional�Tuple�Union)�get_kernel_module_parameter)�IP� IPVersions ^nameserver ([\w.:]+))�tcp�udp�all)�in�outz/etc/resolv.conf�
ip_versionc # � K � t t d� � 5 }|� � � }d d d � � n # 1 swxY w Y n# t $ r g cY S w xY wt � |� � D ]K}|� d� � }t j |� � r | �t j |� � | k s| �|V � �Ld S )N�rb�ascii)
�open�RESOLV_CONF_FILE�read�OSError�DNS_IP_TEMPLATE�findall�decoder � is_valid_ip�type_of)r �f�content�ips �D/opt/imunify360/venv/lib/python3.11/site-packages/im360/utils/net.py�local_dns_from_resolv_confr( ! s � � � ��
�"�D�
)�
)� �Q��f�f�h�h�G� � � � � � � � � � � ���� � � � ��� � � � �� � � ����� �%�%�g�.�.� � ��
�Y�Y�w�
�
��
�>�"� � � ��&�2�:�b�>�>�Z�+G�+G��#������
� s. �A �:� A �>�A �>�A � A�A�returnc # �H K � t j � � � � � D ]y} | D ]t}|j t j k rt
|j � � V � �.|j t j k r1t |j �
dd�� � d � � V � �u�zdS )z8Return a list of IP address assigned to local interfaces�%� )�maxsplitr N) �psutil� net_if_addrs�values�family�socket�AF_INETr �address�AF_INET6r �split)�ipsr& s r' �local_ip_addressesr8 1 s� � � � ��"�$�$�+�+�-�-� H� H��� H� H�B��y�F�N�*�*�!�"�*�-�-�-�-�-�-���f�o�-�-� "�"�*�"2�"2�3��"2�"C�"C�A�"F�G�G�G�G�G��
H�H� H� c � � | t t fv sJ �| t k rd}nd}t d� t j |� � D � � � � S )zZ
Return listening ports for ipv4 services
:param proto: tcp/udp
:return:
�tcp4�udp4c 3 �| K � | ]7}|j d dk �|j t j k �(|j d V � �8dS )r z 127.0.0.1r, N)�laddr�statusr. � CONN_LISTEN)�.0�cs r' � <genexpr>z"listening_ports.<locals>.<genexpr>J sT � � � � � �
�
�G�A�J�+� %� %�!�(�f�6H�*H�*H�
���
�*H�*H�*H�*H�� r9 )�TCP�UDP�setr. �net_connections)�proto�kinds r' �listening_portsrJ = sg � � �S�#�J� � � � � ��|�|������ � � ��'��-�-�� � � � � r9 r, )�maxsizec � � t dd�� � } | dk S # t $ r%}t � |� � Y d}~dS d}~ww xY w)z�
Checks whether ipv6 kernel module is not disabled or absent
https://www.kernel.org/doc/Documentation/networking/ipv6.txt
:return:
�ipv6�disable)� module_name� parameter�0NF)r �
ValueError�logger�warning)rN �es r' �is_ipv6_enabledrV Q sl � � �-� �)�
�
�
��
�#�~� �� � � � ����q���� �u�u�u�u�u����� ���s � �
A�A�Ar& c �h � t � | � ddd�� � dd�� � S )N� �bigT��signedF)�int�
from_bytes�to_bytes�r& s r' �_signed_to_unsignedr` g s6 � � �>�>�
� � �A�u�T� �*�*�E�%� � � � r9 c � � | dk r|}n)t || | dz
z � � � � � }||z
}| |z }d|z dz
}||k r/|dz }| � |d� � }d}t ||z � � �|S )a( Return prefix length from the bitwise netmask.
Copied from ipaddress private method
Args:
ip_int: An integer, the netmask in expanded bitwise format
Returns:
An integer, the prefix length.
Raises:
ValueError: If the input intermingles zeroes & ones
r r, rX rY z&Netmask pattern %r mixes zeroes & ones)�min�
bit_lengthr^ rR ) �ip_int�prefix_maxlength�trailing_zeroes� prefixlen� leading_ones�all_ones�byteslen�details�msgs r' �_prefix_from_maskrm m s� � � ��{�{�*��� � ��w�&�1�*� 5� A� A� C� C�
�
�� !�?�2�I��_�,�L��Y��!�#�H��x� � �#�q�(���/�/�(�E�2�2��6����w� �'�'�'� �r9 �
ip_addressc � � | j dk r*t � | j d d� dd�� � S t | � � S )N� rX rY TrZ )�versionr\ r] �packed)rn s r' �pack_ip_addressrs � s@ � ���Q� � ��~�~�j� /���� 3�U�4�~�H�H�H��:��� r9 r
c �f � t | j � � }t | j � � }||| j fS �N)rs �network_address�netmaskrq )r
�net�masks r' �pack_ip_networkrz � s4 � � �*�4�
5�
5�C� �:�-� .� .�D� ��j�(� (�(r9 rx ry rq c �� � |dk rJt | � � dz } t |� � dz }t |t � � }t | |f� � S t |t � � }t | |f� � S )Nrp �@ )r` rm r r r r )rx ry rq �prefixs r' �unpack_ip_networkr~ � sq � ��!�|�|�!�#�&�&�"�,��"�4�(�(�B�.��"�4��4�4���C��=�)�)�)�"�4��4�4���C��=�)�)�)r9 c �D � t | � � S # t $ r Y d S w xY wru )r
rR r_ s r' �is_netr� � s5 � ���"�~�~� �� � � � ��t�t����s � �
� ru )1�logging�rer2 � functoolsr � ipaddressr r r r r r r
�typingr r r
r r. �defence360agent.utilsr � defence360agent.utils.validater r � getLogger�__name__rS �compile�Mr rD rE �ALL�IN�OUT� IPNetworkr r( r8 rJ rV r\ r` rm rs rz r~ �strr� � r9 r' �<module>r� s� �� ���� � � � �
�
�
�
� � � � � � �� � � � � � � � � � � � � � � � � � 4� 3� 3� 3� 3� 3� 3� 3� 3� 3� 3� 3�
�
�
�
� =� =� =� =� =� =� 8� 8� 8� 8� 8� 8� 8� 8� �� �8� $� $�� �"�*� 6���=�=��#�
��S�#�
���C� �+�{�*� +� �%� �
�
�8�I�+>�
�
�
�
� H�H�U�;� �+C�%D� E� H� H� H� H�� � �( ��1����� � ���*�C� � � � � � � �> ��k�;�&>� ?� � � � �)��k�;�.�/�)�
�3��S�=��)� )� )� )�*�3� *�c� *�C� *�I� *� *� *� *��s� �u�T�;� � ;�<� � � � � � r9