Team Ai
Datasetpublic

codekingpro/portable-devtools

sourceHugging Faceupdated 5mo agoView on Hugging Face
1likes14kdownloads
dns.cpython-313.pyc54 linesDownload Raw Back to __pycache__
1�

2��j���SSKrSSKrSSKJr SSKJr SSKJr SSKJr SSKJ	r3 SSKJr SSK
Jr SS	K
Jr SS4K
Jr SSKJr SSKJr \R*"S
5r\"SS\R.55r\"SS\R.55r\"SS\R.55rS\R6S\SS\4Sjr"SS\R<5rg)�N)�	dataclass)�List)�Literal)�dns)�flow)�response_codes)�commands)�events)�layer)�Context)�expect�!Hc�8�\rSrSr%Sr\R\S'Srg)�DnsRequestHook�z 5A DNS query has been received.6r�N�	�__name__�7__module__�__qualname__�__firstlineno__�__doc__r�DNSFlow�__annotations__�__static_attributes__r��`D:\code\apps\devtools\python\user_packages\Python313\site-packages\mitmproxy/proxy/layers/dns.pyrr�����+�+�rrc�8�\rSrSr%Sr\R\S'Srg)�DnsResponseHook�z*8A DNS response has been received or set.9rrNrrrrr r rrr c�8�\rSrSr%Sr\R\S'Srg)�DnsErrorHook�%z10A DNS error has occurred.11rrNrrrrr#r#%rrr#�message�transport_protocol)�tcp�udp�returnc�p�URnUS:Xa#[R"S[U55U-$U$)Nr'r)�packed�struct�pack�len)r%r&r+s   r�pack_messager/.s4���^�^�F��U�"��{�{�4��V��-��6�6��
rc��^�\rSrSr%Sr\\\R4\	S'\12\	S'\13\	S'S\4U4SjjrS\RS	\RS14\RS4SjrS\RS	\RS15\RS4S
jrS\RS\S16\RS4SjrS\S\S17\\R4Sjr\"\R45S18\RS4Sj5r\"\R8\R:5S\R<S19\RS4Sj5r\"\R8\R:5S20\RS4Sj5r \r!Sr"U=r#$)�DNSLayer�8z+21Layer that handles resolving DNS queries.22�flows�req_buf�resp_buf�contextc�n>�[TU]U5 0Ul[5Ul[5Ulg�N)�super�__init__r3�	bytearrayr4r5)�selfr6�	__class__s  �rr:�DNSLayer.__init__As)���
����!���23� �{���!���
rr�msgr)Nc#�T# �X!l[U5v� UR(a$URXR5Shv�N gUR(a.URXRR5Shv�N gURRR(dURUS5Shv�N gURRR(dW[R"URR5v�nU(a$URU[U55Shv�N g[URURR 5n[R""URRU5v� gGNYGNN�Ne7f)Nz;No hook has set a response and there is no upstream server.)�requestr�response�handle_response�error�handle_errorr?r6�server�address�	connectedr	�OpenConnection�strr/�server_connr&�SendData)r<rr?�errr+s     r�handle_request�DNSLayer.handle_requestGs)������T�"�"��=�=��+�+�D�-�-�@�@�@�
�Z�Z��(�(��z�z�~�~�>�>�>����$�$�,�,��(�(��S��
�
��<�<�&�&�0�0�$�3�3�D�L�L�4G�4G�H�H���#�0�0��s�3�x�@�@�@��!�$�,�,��0@�0@�0S�0S�T�F��#�#�D�L�L�$7�$7��@�@�
A�>�
�A�sJ�AF(�F�>F(�F!�?F(�F$�A;F(�F&�AF(�!F(�$F(�&F(c#�# �X!l[U5v� UR(aX[URURR5n[24R"URRU5v� gg7fr8)	rBr r/�client_connr&r	rLr6�client)r<rr?r+s    rrC�DNSLayer.handle_response^s\����
��d�#�#��=�=�!�$�-�-��1A�1A�1T�1T�U�F��#�#�D�L�L�$7�$7��@�@��s�A>BrMc#�B# �[R"U5Ul[U5v� URR[R5n[R"URR[X1RR55v� g7fr8)�mflow�ErrorrDr#rA�failr�SERVFAILr	rLr6rRr/rQr&)r<rrM�servfails    rrE�DNSLayer.handle_errorgsn����[�[��%��25��4� � ��<�<�$�$�^�%<�%<�=������L�L�����#3�#3�#F�#F�G�26�	27�s�BB�data�from_clientc�@�/nU(aUROURnURRRS:XaBUR[RRU[R"5S95 U$URRRS:Xa�URU5 [U5nSnXV-28[R:aO�[RXF5unU[R-
nUS:Xa[R "S5eXV-29U:aU[R-nOU[#XFXv-5nXg-
nUR[RRU[R"5S95 M�USU2	U$)Nr()�	timestampr'rz#Message length field cannot be zero)r4r5r6rRr&�appendr�30DNSMessage�unpack�time�extendr.�
_LENGTH_LABEL�size�unpack_fromr,rD�bytes)r<r[r\�msgs�bufre�offset�
expected_sizes        r�unpack_message�DNSLayer.unpack_messagepsV��%'��)�d�l�l�t�}�}���<�<���1�1�U�:��K�K����-�-�d�d�i�i�k�-�J�K�.��-�\�\�
 �
 �
3�
3�u�
<��J�J�t���s�8�D��F���=�=�#5�#5�5��#0�#<�#<�S�#I� ���-�,�,�,�� �A�%� �,�,�'L�M�M��=�=�0��m�0�0�0�F���S�-�*@�A�B���'�����C�N�N�1�1�$�$�)�)�+�1�N�O�� �G�V�G���rc#�F# �URUlSShv�N gN7f�Nr)�state_query�
_handle_event�r<�_s  r�state_start�DNSLayer.state_start�s���!�-�-����
�
�s�!��!�eventc#� # �[U[R5(deURURR31Ln[U[R5(a~/nURURU5nUHXnURURnU(aURXT5Shv�N M?UR!XT5Shv�N MZ g[U[R05(a�U(aURROURR32nUR2(a[&R*"U5v� UR,UlURR55H33nSUlM g[9SU35e![aZ [R"URR34URRSS9nXPRUR'GNff=fGNOGN7!["R$adn[&R("URSU35v� [&R*"UR5v� UR,UlSnAgSnAff=f7f)NT)�livez sent an invalid message: FzUnexpected event: )�35isinstancer36�ConnectionEvent�37connectionr6rR�DataReceivedrlr[r3�id�KeyErrorrrrFrNrCr,rDr	�Log�CloseConnection�38state_donerq�ConnectionClosedrH�valuesrx�AssertionError)r<rvr\rhr?r�e�39other_conns        rrp�DNSLayer.state_query�s�����%��!7�!7�8�8�8�8��&�&�$�,�,�*=�*=�=���e�V�0�0�1�1�)+�D�
C��*�*�5�:�:�{�C�� �C�2�#�z�z�#�&�&�1��#�#'�#6�#6�t�#A�A�A�#'�#7�#7��#B�B�B� ���v�6�6�
7�
7�0;����,�,����AT�AT�J��#�#��.�.�z�:�:�!%���D���40�41�)�)�+��!��	�,�!�#5�e�W�!=�>�>��'$�2�"�{�{� �L�L�/�/����1D�1D�4� ��.2�42�43�3�6�6�*�	2��B�B��!�<�<�
5��l�l�e�&6�&6�%7�7Q�RS�QT�#U�V�V��.�.�u�/?�/?�@�@�%)�_�_��"�"��
5�su�A%J�(H�J�44F&�#J�>H
�?J�H�CJ�&A H45�J�	H46�47J�J�J�'AJ�J�J�Jc#�$# �SShv�N gN7frorrrs  rr��DNSLayer.state_done�s����
�
�s���)rqr3r4r5)$rrrrr�dict�intrrrr;rr:r`r�CommandGeneratorrNrCrJrErg�boolrrlr
r48�Startrtr|r��Eventrpr�rqr�
__classcell__)r=s@rr1r18s�������S�[�[� �!�!�
����$��$�A��K�K�A�&)�n�n�A�	�	�	��	%�A�.A��K�K�A�&)�n�n�A�	�	�	��	%�A�49����50�3�51�5�;Q�;Q�RV�;W�52��5��t���S�^�^�@T��>�F�L�L���� 6� 6�t� <�����F����!8�!8�9�#?����#?�%�2H�2H��2N�#?�:�#?�J�F����!8�!8�9��u�5�5�d�;��:�� �Mrr1) r,rb�dataclassesr�typingrr�	mitmproxyrrrU�mitmproxy.net.dnsr�mitmproxy.proxyr	r53r�mitmproxy.proxy.contextr�mitmproxy.proxy.utilsr
�Structrd�	StartHookrr r#r`rgr/�Layerr1rrr�<module>r�s���
��!����#�,�$�"�!�+�(��
�
�d�#�
���X�'�'������h�(�(������8�%�%�����
�^�^��18��1F��54��F �u�{�{�F r
codekingpro/portable-devtools · Team Ai