�
5�Dg� �
�L � d dl Z d dlZd dlZd dlZd dlmZmZmZmZ ddl m
Z
dZ dZ dZ
dZej ej ej ej ej ej ej ej ej ej ej ej d� Zej ej ej ej ej d �Z! G d
� d � � Z"dS ) � N)�Any�List�Optional�Set� )�errorz3.0.0)� DNSResolverr � ) �A�AAAA�ANY�CAA�CNAME�MX�NAPTR�NS�PTR�SOA�SRV�TXT)�IN�CHAOS�HS�NONEr
c � � e Zd Z d deee deej deddfd�Z e
de j fd�� � Z
e
j dee ddfd �� � Z
ed
ej d ed eddfd
�� � Zd dedededej fd�Zdedej dej fd�Zdedej fd�Zd d�Zdedededdfd�Zdededdfd �Zd d �Z dS )!r N� nameservers�loop�kwargs�returnc � � |pt j � � | _ | j �J �|� dd � � t j dd| j i|��| _ |r|| _ t � � | _
t � � | _ d | _ d S )N�
sock_state_cb� )
�asyncio�get_event_loopr �pop�pycares�Channel�_sock_state_cb�_channelr �set� _read_fds�
_write_fds�_timer)�selfr r r s �o/builddir/build/BUILD/imunify360-venv-2.4.0/opt/imunify360/venv/lib/python3.11/site-packages/aiodns/__init__.py�__init__zDNSResolver.__init__/ s� � � �4�G� 2� 4� 4�� ��y�$�$�$��
�
�?�D�)�)�)� ��T�T�d�6I�T�V�T�T��
� � +� *�D� � ����� �%�%����� � � � c � � | j j S �N�r) �servers�r. s r/ r zDNSResolver.nameservers< s
� ��}�$�$r1 �valuec � � || j _ d S r3 r4 )r. r7 s r/ r zDNSResolver.nameservers@ s � � %��
� � � r1 �fut�result�errornoc �� � | � � � rd S |�G| � t j |t j � |� � � � � � d S | � |� � d S r3 )� cancelled�
set_exceptionr �DNSErrorr& �errno�strerror�
set_result)r9 r: r; s r/ � _callbackzDNSResolver._callbackD sk � � �=�=�?�?� � �F� � � � � �e�n�W�g�m�6L�6L�W�6U�6U� V� V� W� W� W� W� W� �N�N�6� "� "� "� "� "r1 �host�qtype�qclassc � � t | }n0# t $ r# t d� |� � � � �w xY w|�? t | }n0# t $ r# t d� |� � � � �w xY wt j | j �� � }t j | j
|� � }| j � ||||�� � |S )Nzinvalid query type: {}zinvalid query class: {}�r )� query_class)
�query_type_map�KeyError�
ValueError�format�query_class_mapr# �Futurer � functools�partialrC r) �query)r. rD rE rF r9 �cbs r/ rR zDNSResolver.queryM s� � � E�"�5�)�E�E��� E� E� E� � 5� <� <�U� C� C�D�D� D� E���� � �
K�(��0�����
K�
K�
K� �!:�!A�!A�&�!I�!I�J�J�J�
K���� �n�$�)�,�,�,��
�
�t�~�s�
3�
3�� �
���D�%����@�@�@��
s �
�-=�
A �-A>�familyc � � t j | j �� � }t j | j |� � }| j � |||� � |S �NrH )r# rO r rP rQ rC r) �
gethostbyname)r. rD rT r9 rS s r/ rW zDNSResolver.gethostbyname] sI � ��n�$�)�,�,�,��
�
�t�~�s�
3�
3�� �
�#�#�D�&�"�5�5�5��
r1 �namec � � t j | j �� � }t j | j |� � }| j � ||� � |S rV )r# rO r rP rQ rC r) �
gethostbyaddr)r. rX r9 rS s r/ rZ zDNSResolver.gethostbyaddrc sG � ��n�$�)�,�,�,��
�
�t�~�s�
3�
3�� �
�#�#�D�"�-�-�-��
r1 c �8 � | j � � � d S r3 )r) �cancelr6 s r/ r\ zDNSResolver.canceli s � � �
� � � � � � � r1 �fd�readable�writablec �� � |s|r�|rA| j � || j |t � � | j � |� � |rA| j � || j |t � � | j � |� � | j �'| j �
d| j � � | _ d S d S || j v r4| j � |� � | j �
|� � || j v r4| j � |� � | j � |� � | j s0| j s+| j �&| j � � � d | _ d S d S d S d S �Ng �?)r �
add_reader�
_handle_event�READr+ �add�
add_writer�WRITEr, r- �
call_later� _timer_cb�discard�
remove_reader�
remove_writerr\ )r. r] r^ r_ s r/ r( zDNSResolver._sock_state_cbl s� � � � #�x� #��
'�� �$�$�R��);�R��F�F�F���"�"�2�&�&�&��
(�� �$�$�R��);�R��G�G�G���#�#�B�'�'�'��{�"� "�i� 2� 2�3��� G� G�� � � � #�"� �T�^�#�#���&�&�r�*�*�*�� �'�'��+�+�+��T�_�$�$���'�'��+�+�+�� �'�'��+�+�+��>�
#�$�/�
#�d�k�>U�� �"�"�$�$�$� "�� � � �
#�
#�
#�
#�>U�>Ur1 �eventc � � t j }t j }|t k r|}n
|t k r|}| j � ||� � d S r3 )r& �ARES_SOCKET_BADrd rg r) �
process_fd)r. r] rm �read_fd�write_fds r/ rc zDNSResolver._handle_event� sN � ��)���*�� �D�=�=��G�G�
�e�^�^��H� �
� � ��(�3�3�3�3�3r1 c �� � | j s| j rV| j � t j t j � � | j � d| j � � | _ d S d | _ d S ra )
r+ r, r) rp r& ro r rh ri r- r6 s r/ ri zDNSResolver._timer_cb� s] � � �>� �T�_� � �M� $� $�W�%<�g�>U� V� V� V� �)�.�.�s�D�N�C�C�D�K�K�K� �D�K�K�Kr1 )NNr3 )r N) �__name__�
__module__� __qualname__r r �strr# �AbstractEventLoopr r0 �propertyr& r' r �setter� staticmethodrO �intrC rR �socket�
AddressFamilyrW rZ r\ �boolr( rc ri r" r1 r/ r r . s
� � � � � �:>�=A� � �H�T�#�Y�$7� � �� 9�:� � � �#'� � � � � �%�W�_� %� %� %� �X�%� ��&��c�� &�t� &� &� &� ��&� �#�w�~� #�s� #�S� #�T� #� #� #� �\�#�� �#� �c� �3� ��� � � � � �#� �v�/C� ��� � � � � �#� �'�.� � � � � � � � �#�� #�� #�� #�� #� #� #� #�04�� 4�C� 4�D� 4� 4� 4� 4� � � � � � r1 r )#r# rP r& r} �typingr r r r � r � __version__�__all__rd rg � QUERY_TYPE_A�QUERY_TYPE_AAAA�QUERY_TYPE_ANY�QUERY_TYPE_CAA�QUERY_TYPE_CNAME�
QUERY_TYPE_MX�QUERY_TYPE_NAPTR�
QUERY_TYPE_NS�QUERY_TYPE_PTR�QUERY_TYPE_SOA�QUERY_TYPE_SRV�QUERY_TYPE_TXTrJ �QUERY_CLASS_IN�QUERY_CLASS_CHAOS�QUERY_CLASS_HS�QUERY_CLASS_NONE�QUERY_CLASS_ANYrN r r" r1 r/ �<module>r� sP �� ���� � � � � ����
�
�
�
�� � � � � � � � � � � � � � � � � � � �
"�� �� �� #� 0� #� 3� #� 2� #� 2� #� 4� #� 1� #� 4� #� 1� #� 2� #� 2� #� 2� #� 2�
�
�� %� 3� $� 6� $� 3� $� 5� $� 4� � ��d � d � d � d � d � d � d � d � d � d r1